[tvm] branch main updated: Fix bug that disabled cuda integer dot product. (#12099)

[ma...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

masahi pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new c0f4450958 Fix bug that disabled cuda integer dot product. (#12099)
c0f4450958 is described below

commit c0f445095813d7b1ceaf399e697efe02fc7278f3
Author: Josh Fromm <jw...@octoml.ai>
AuthorDate: Thu Jul 14 17:22:17 2022 -0700

    Fix bug that disabled cuda integer dot product. (#12099)
---
 python/tvm/target/target.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/tvm/target/target.py b/python/tvm/target/target.py
index 7c1e55c39e..2518527083 100644
--- a/python/tvm/target/target.py
+++ b/python/tvm/target/target.py
@@ -197,7 +197,7 @@ class Target(Object):
     def supports_integer_dot_product(self):
         if self.attrs.get("supports_integer_dot_product", []):
             return bool(self.attrs["supports_integer_dot_product"])
-        if self.kind == "cuda":
+        if self.kind.name == "cuda":
             sm_version = int(self.arch.split("_")[1])
             if sm_version >= 61:
                 return True