You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by GitBox <gi...@apache.org> on 2021/10/29 21:06:05 UTC

[GitHub] [tvm] masahi commented on a change in pull request #9399: [CUTLASS] Fix hardcoded include path and logic for profile_all = False case

masahi commented on a change in pull request #9399:
URL: https://github.com/apache/tvm/pull/9399#discussion_r739531525



##########
File path: python/tvm/contrib/cutlass/build.py
##########
@@ -16,11 +16,17 @@
 # under the License.
 # pylint: disable=invalid-name
 """Driver for partitioning and building a Relay module for CUTLASS offload."""
+import os
 import tvm
 from tvm import runtime, relay
 from .gen_gemm import CutlassGemmProfiler
 
 
+def _get_cutlass_path():
+    tvm_root = os.path.join(os.path.dirname(os.path.realpath(__file__)), "../../../../")
+    return os.path.join(tvm_root, "3rdparty/cutlass")

Review comment:
       hmm right... I don't have a good solution for that problem.
   
   Can we say we don't support cutlass on such environments?




-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org