[CuTeDSL] Make editable installs use exact runtime companion wheels

.gitignore

@@ -2,3 +2,11 @@
 __pycache__/
 cutlass_library.egg-info/
 /build*
+
+# CuTe DSL editable-install payload copied from nvidia-cutlass-dsl-libs-*.
+python/CuTeDSL/VERSION.EDITABLE
+python/CuTeDSL/cutlass/_mlir/
+python/CuTeDSL/cutlass/base_dsl/py.typed
+python/CuTeDSL/cutlass/cute/py.typed
+python/CuTeDSL/lib/
+python/CuTeDSL/nvidia_cutlass_dsl.egg-info/

python/CuTeDSL/cutlass/base_dsl/compiler.py

@@ -18,16 +18,13 @@
 from typing import Any
 import collections.abc
 import os
-import sys
 import inspect
+import importlib
 import types
 from .common import DSLRuntimeError
 from .utils.logger import log
 from .env_manager import EnvironmentVarManager
 
-_SCRIPT_PATH = os.path.dirname(os.path.abspath(__file__))
-sys.path.append(_SCRIPT_PATH)
-
 from .._mlir import ir
 
 
@@ -477,11 +474,11 @@ def enable_tvm_ffi(self) -> bool:
         ret = self.options[EnableTVMFFI].value
         if ret:
             try:
-                import tvm_ffi
-            except ModuleNotFoundError:
+                _ = importlib.import_module("tvm_ffi")
+            except Exception as e:
                 raise DSLRuntimeError(
                     "TVM FFI is not installed, please install it via `pip install apache-tvm-ffi`"
-                )
+                ) from e
         return ret
 
     def to_str(self) -> str:
@@ -521,7 +518,6 @@ def _get_compile_option_from_str(option_str: str) -> type[CompileOption]:
         return mapping[option_str]
 
     import argparse
-    import shlex
 
     parser = argparse.ArgumentParser()
     parser.add_argument("--opt-level", nargs="?", type=int, default=3)

python/CuTeDSL/cutlass/cute/algorithm.py

@@ -42,7 +42,6 @@
     CopyAtom,
     make_atom,
     _normalize_variadic_tensor_operand,
-    copy_atom_call,
 )
 from .nvgpu.common import (
     CacheEvictionPriority,