[LANG/PASS] InjectVirtualThread (#38)
Showing
- include/tvm/ir.h 24 additions, 0 deletionsinclude/tvm/ir.h
- include/tvm/ir_mutator.h 1 addition, 0 deletionsinclude/tvm/ir_mutator.h
- include/tvm/ir_pass.h 20 additions, 0 deletionsinclude/tvm/ir_pass.h
- python/tvm/build.py 2 additions, 0 deletionspython/tvm/build.py
- src/api/api_pass.cc 2 additions, 0 deletionssrc/api/api_pass.cc
- src/arithmetic/canonical.cc 2 additions, 1 deletionsrc/arithmetic/canonical.cc
- src/codegen/codegen_c.cc 2 additions, 2 deletionssrc/codegen/codegen_c.cc
- src/codegen/codegen_cuda.cc 12 additions, 0 deletionssrc/codegen/codegen_cuda.cc
- src/codegen/codegen_cuda.h 6 additions, 0 deletionssrc/codegen/codegen_cuda.h
- src/pass/inject_virtual_thread.cc 419 additions, 0 deletionssrc/pass/inject_virtual_thread.cc
- src/pass/ir_mutator.cc 12 additions, 10 deletionssrc/pass/ir_mutator.cc
- src/pass/lift_allocate.cc 96 additions, 0 deletionssrc/pass/lift_allocate.cc
- src/pass/storage_flatten.cc 20 additions, 69 deletionssrc/pass/storage_flatten.cc
- src/runtime/thread_storage_scope.h 5 additions, 1 deletionsrc/runtime/thread_storage_scope.h
- src/schedule/schedule_ops.cc 11 additions, 2 deletionssrc/schedule/schedule_ops.cc
- tests/python/unittest/test_pass_virtual_thread.py 28 additions, 0 deletionstests/python/unittest/test_pass_virtual_thread.py
Loading
Please register or sign in to comment