add gfx906 bc (#3808)

c87ace7e · Peter Yeh · masahi · aa7bf34c · c87ace7e
Commit c87ace7e authored Aug 20, 2019 by Peter Yeh Committed by masahi Aug 21, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 additions and 0 deletions

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

No files found.
--- a/python/tvm/contrib/rocm.py
+++ b/python/tvm/contrib/rocm.py
@@ -130,6 +130,7 @@ def callback_rocm_bitcode_path(rocdl_dir="/opt/rocm/lib/"):
        "oclc_finite_only_on.amdgcn.bc",
        "oclc_isa_version_803.amdgcn.bc",
        "oclc_isa_version_900.amdgcn.bc",
+        "oclc_isa_version_906.amdgcn.bc",
        "oclc_unsafe_math_off.amdgcn.bc",
        "oclc_unsafe_math_on.amdgcn.bc"
    ]