Fix cutlass and tk install

pytorch-labs · Nov 5, 2024 · c85f689 · c85f689
1 parent 92be7af
commit c85f689
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/install.py b/install.py
@@ -53,7 +53,7 @@ def test_fbgemm():
 
 
 def install_cutlass():
-    from utils.cutlass_kernels.install import install_colfax_cutlass
+    from tools.cutlass_kernels.install import install_colfax_cutlass
 
     install_colfax_cutlass()
 
@@ -84,7 +84,7 @@ def install_liger():
 
 
 def install_tk():
-    from utils.tk.install import install_tk
+    from tools.tk.install import install_tk
 
     install_tk()
 

diff --git a/tools/cutlass_kernels/install.py b/tools/cutlass_kernels/install.py
@@ -11,7 +11,7 @@
 FBGEMM_PATH = REPO_PATH.joinpath("submodules", "FBGEMM", "fbgemm_gpu")
 FBGEMM_CUTLASS_PATH = FBGEMM_PATH.parent.joinpath("external", "cutlass")
 COLFAX_CUTLASS_PATH = REPO_PATH.joinpath("submodules", "cutlass-kernels")
-COLFAX_CUTLASS_TRITONBENCH_PATH = REPO_PATH.joinpath("utils", "cutlass_kernels")
+COLFAX_CUTLASS_TRITONBENCH_PATH = REPO_PATH.joinpath("tools", "cutlass_kernels")
 
 TORCH_BASE_PATH = Path(torch.__file__).parent
 

diff --git a/tools/tk/install.py b/tools/tk/install.py
@@ -11,7 +11,7 @@
 REPO_PATH = Path(os.path.abspath(__file__)).parent.parent.parent
 TK_PATH = REPO_PATH.joinpath("submodules", "ThunderKittens")
 TK_PYUTILS_PATH = TK_PATH.joinpath("src", "common", "pyutils")
-TRITONBENCH_TK_PATH = REPO_PATH.joinpath("utils", "tk")
+TRITONBENCH_TK_PATH = REPO_PATH.joinpath("tools", "tk")
 TORCH_BASE_PATH = Path(torch.__file__).parent
 CONDA_PREFIX = os.environ.get("CONDA_PREFIX", None)
 PYTHON_VERSION = f"python{sys.version_info[0]}.{sys.version_info[1]}"