Skip to content

T
tic
Commit beb1c252 by Peter Yeh Committed by masahi
Options

add bc for gfx1010 (#3984)

parent 4ba911a7
Showing with 2 additions and 1 deletions
python/tvm/contrib/rocm.py
...@@ -132,7 +132,8 @@ def callback_rocm_bitcode_path(rocdl_dir="/opt/rocm/lib/"): ...@@ -132,7 +132,8 @@ def callback_rocm_bitcode_path(rocdl_dir="/opt/rocm/lib/"):
"oclc_isa_version_900.amdgcn.bc", "oclc_isa_version_900.amdgcn.bc",
"oclc_isa_version_906.amdgcn.bc", "oclc_isa_version_906.amdgcn.bc",
"oclc_unsafe_math_off.amdgcn.bc", "oclc_unsafe_math_off.amdgcn.bc",
"oclc_unsafe_math_on.amdgcn.bc" "oclc_unsafe_math_on.amdgcn.bc",
"oclc_wavefrontsize64_on.amdgcn.bc"
] ]
paths = [join(rocdl_dir, bitcode) for bitcode in bitcode_files] paths = [join(rocdl_dir, bitcode) for bitcode in bitcode_files]
return convert([path for path in paths if exists(path)]) return convert([path for path in paths if exists(path)])
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment