add gfx906 bc (apache#3808)

kimishpatel · Aug 21, 2019 · c87ace7 · c87ace7
1 parent aa7bf34
commit c87ace7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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"
     ]