add bc for gfx1010 (#3984)

beb1c252 · Peter Yeh · masahi · 4ba911a7 · beb1c252
Commit beb1c252 authored Sep 20, 2019 by Peter Yeh Committed by masahi Sep 21, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletions

python/tvm/contrib/rocm.py
+2 -1

No files found.
--- a/python/tvm/contrib/rocm.py
+++ b/python/tvm/contrib/rocm.py
@@ -132,7 +132,8 @@ def callback_rocm_bitcode_path(rocdl_dir="/opt/rocm/lib/"):
        "oclc_isa_version_900.amdgcn.bc",
        "oclc_isa_version_906.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]
    return convert([path for path in paths if exists(path)])