-
- Downloads
[ROCM] Working math function support for ROCm backend, a bug fix in LLVM based codegen (#570)
* added math function support * bug fix extern func call in llvm based codegen lint fix fix build bug fix extern func call in llvm based codegen * moved rocm bitcodes detection to python
Showing
- python/tvm/contrib/rocm.py 31 additions, 1 deletionpython/tvm/contrib/rocm.py
- src/codegen/llvm/codegen_amdgpu.cc 18 additions, 0 deletionssrc/codegen/llvm/codegen_amdgpu.cc
- src/codegen/llvm/codegen_llvm.cc 2 additions, 3 deletionssrc/codegen/llvm/codegen_llvm.cc
- src/codegen/llvm/intrin_rule_rocm.cc 48 additions, 0 deletionssrc/codegen/llvm/intrin_rule_rocm.cc
Loading
Please register or sign in to comment