projects / platform / upstream / tvm.git / commitdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | patch | inline | side by side (parent: aa7bf34)
add gfx906 bc (#3808)
authorPeter Yeh <petrex@users.noreply.github.com>
Wed, 21 Aug 2019 01:57:01 +0000 (18:57 -0700)
committermasahi <masahi129@gmail.com>
Wed, 21 Aug 2019 01:57:01 +0000 (10:57 +0900)
python/tvm/contrib/rocm.py patch | blob | history

diff --git a/python/tvm/contrib/rocm.py b/python/tvm/contrib/rocm.py
index d81ca56..5db8a38 100644 (file)
--- 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"
     ]
Domain: Machine Learning / ML Framework;