-
- Downloads
[RUNTIME][TRACE] Support trace primitive. (#1973)
Add trace call expr to allow trace Tensor data at the runtime. By default the default handler is enabled which prints a tracing data to stdout, otherwise user should specify a call_back as global_function (aka @tvm.register_func). The issue is related to: https://discuss.tvm.ai/t/idea-trace-expression/945
Showing
- include/tvm/ir.h 30 additions, 0 deletionsinclude/tvm/ir.h
- python/tvm/intrin.py 36 additions, 0 deletionspython/tvm/intrin.py
- src/codegen/llvm/codegen_cpu.cc 70 additions, 29 deletionssrc/codegen/llvm/codegen_cpu.cc
- src/codegen/llvm/codegen_cpu.h 7 additions, 0 deletionssrc/codegen/llvm/codegen_cpu.h
- src/pass/lower_tvm_builtin.cc 52 additions, 1 deletionsrc/pass/lower_tvm_builtin.cc
- tests/python/unittest/test_runtime_packed_func.py 179 additions, 0 deletionstests/python/unittest/test_runtime_packed_func.py
Loading
Please register or sign in to comment