Skip to content

Commit 0ea22df

Browse files
petrexwweic
authored andcommitted
add bc for gfx1010 (apache#3984)
1 parent ed5f74e commit 0ea22df

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

python/tvm/contrib/rocm.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,8 @@ def callback_rocm_bitcode_path(rocdl_dir="/opt/rocm/lib/"):
132132
"oclc_isa_version_900.amdgcn.bc",
133133
"oclc_isa_version_906.amdgcn.bc",
134134
"oclc_unsafe_math_off.amdgcn.bc",
135-
"oclc_unsafe_math_on.amdgcn.bc"
135+
"oclc_unsafe_math_on.amdgcn.bc",
136+
"oclc_wavefrontsize64_on.amdgcn.bc"
136137
]
137138
paths = [join(rocdl_dir, bitcode) for bitcode in bitcode_files]
138139
return convert([path for path in paths if exists(path)])

0 commit comments

Comments
 (0)