[LANG] Generalize compute to tensor region (#1476)
Showing
- 3rdparty/dmlc-core 1 addition, 1 deletion3rdparty/dmlc-core
- include/tvm/expr.h 2 additions, 0 deletionsinclude/tvm/expr.h
- include/tvm/operation.h 70 additions, 2 deletionsinclude/tvm/operation.h
- include/tvm/tensor_intrin.h 53 additions, 0 deletionsinclude/tvm/tensor_intrin.h
- python/tvm/api.py 29 additions, 10 deletionspython/tvm/api.py
- python/tvm/tensor.py 12 additions, 0 deletionspython/tvm/tensor.py
- python/tvm/tensor_intrin.py 26 additions, 2 deletionspython/tvm/tensor_intrin.py
- src/api/api_lang.cc 20 additions, 0 deletionssrc/api/api_lang.cc
- src/lang/tensor.cc 36 additions, 8 deletionssrc/lang/tensor.cc
- src/op/compute_op.cc 35 additions, 0 deletionssrc/op/compute_op.cc
- src/op/compute_op.h 16 additions, 1 deletionsrc/op/compute_op.h
- src/op/tensor_compute_op.cc 361 additions, 0 deletionssrc/op/tensor_compute_op.cc
- src/op/tensorize.cc 0 additions, 45 deletionssrc/op/tensorize.cc
- src/pass/arg_binder.cc 3 additions, 1 deletionsrc/pass/arg_binder.cc
- src/schedule/schedule_dataflow_rewrite.cc 191 additions, 61 deletionssrc/schedule/schedule_dataflow_rewrite.cc
- tests/python/unittest/test_lang_tensor.py 74 additions, 0 deletionstests/python/unittest/test_lang_tensor.py
- tests/python/unittest/test_schedule_schedule_ops.py 130 additions, 0 deletionstests/python/unittest/test_schedule_schedule_ops.py
Loading
Please register or sign in to comment