[Dev] Remove unnecessary python dependencies

install_cpu.sh

@@ -7,6 +7,7 @@ echo "Starting installation script..."
 
 # Step 1: Install Python requirements
 echo "Installing Python requirements from requirements.txt..."
+pip install -r requirements-build.txt
 pip install -r requirements.txt
 if [ $? -ne 0 ]; then
     echo "Error: Failed to install Python requirements."

install_cuda.sh

@@ -7,6 +7,7 @@ echo "Starting installation script..."
 
 # Step 1: Install Python requirements
 echo "Installing Python requirements from requirements.txt..."
+pip install -r requirements-build.txt
 pip install -r requirements.txt
 if [ $? -ne 0 ]; then
     echo "Error: Failed to install Python requirements."

install_rocm.sh

@@ -3,9 +3,17 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 
+echo "Starting installation script..."
+
 # install requirements
+pip install -r requirements-build.txt
 pip install -r requirements.txt
-
+if [ $? -ne 0 ]; then
+    echo "Error: Failed to install Python requirements."
+    exit 1
+else
+    echo "Python requirements installed successfully."
+fi
 # determine if root
 USER_IS_ROOT=false
 if [ "$EUID" -eq 0 ]; then

pyproject.toml

@@ -3,7 +3,6 @@ requires = [
     "cmake>=3.26",
     "packaging",
     "setuptools>=61",
-    "setuptools-scm>=8.0",
     "wheel",
 ]
 build-backend = "setuptools.build_meta"

requirements-build.txt

@@ -0,0 +1,6 @@
+# Should be mirrored in pyproject.toml
+cmake>=3.26
+packaging
+setuptools>=61
+torch
+wheel

requirements-dev.txt

@@ -28,7 +28,6 @@ cloudpickle
 ml_dtypes
 psutil
 scipy
-tornado
 torch
 thefuzz
 tabulate

requirements-test.txt

@@ -27,8 +27,6 @@ attrs
 cloudpickle
 ml_dtypes
 psutil
-scipy
-tornado
 torch
 thefuzz
 tabulate

requirements.txt

@@ -1,26 +1,11 @@
-# build requirements
-cmake>=3.26
 # runtime requirements
-cffi
-cpplint
 Cython
 decorator
-docutils
-dtlib
 numpy>=1.23.5
-pytest>=6.2.4
-pytest_xdist>=2.2.1
-packaging>=21.0
-PyYAML
 tqdm>=4.62.3
 typing_extensions>=4.10.0
-requests
 attrs
 cloudpickle
 ml_dtypes
 psutil
-scipy
-tornado
 torch
-thefuzz
-tabulate

tilelang/tools/plot_layout.py

@@ -132,6 +132,18 @@ def plot_layout(layout: T.Layout,
     plt.xticks([])  # Remove x-axis ticks
     plt.yticks([])  # Remove y-axis ticks
 
+    legend_patches = [
+        patches.Patch(color='black', label="T: Thread ID"),
+        patches.Patch(color='black', label="L: Local ID")
+    ]
+    ax.legend(
+        handles=legend_patches,
+        loc="upper right",
+        fontsize=font_size - 4,
+        frameon=False,
+        bbox_to_anchor=(1.0, 1.12),
+        ncols=2)
+
     # Create the output directory if it does not exist
     tmp_directory = pathlib.Path(save_directory)
     if not os.path.exists(tmp_directory):