We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ed5f74e commit 0ea22dfCopy full SHA for 0ea22df
python/tvm/contrib/rocm.py
@@ -132,7 +132,8 @@ def callback_rocm_bitcode_path(rocdl_dir="/opt/rocm/lib/"):
132
"oclc_isa_version_900.amdgcn.bc",
133
"oclc_isa_version_906.amdgcn.bc",
134
"oclc_unsafe_math_off.amdgcn.bc",
135
- "oclc_unsafe_math_on.amdgcn.bc"
+ "oclc_unsafe_math_on.amdgcn.bc",
136
+ "oclc_wavefrontsize64_on.amdgcn.bc"
137
]
138
paths = [join(rocdl_dir, bitcode) for bitcode in bitcode_files]
139
return convert([path for path in paths if exists(path)])
0 commit comments