Moved common requirements from docker and setup.py to file

docker/install/install_python_packages.sh

@@ -23,11 +23,8 @@ set -u
 CUDA_VERSION=${1:-cu128}
 
 pip3 install torch --index-url https://download.pytorch.org/whl/${CUDA_VERSION}
-pip3 install requests responses ninja pytest numpy scipy build nvidia-ml-py cuda-python einops nvidia-nvshmem-cu12
-pip3 install click
-pip3 install 'apache-tvm-ffi==0.1.0b15'
-pip3 install nvidia-cutlass-dsl
-pip3 install 'nvidia-cudnn-frontend>=1.13.0'
+pip3 install -r requirements.txt
+pip3 install responses pytest scipy build cuda-python nvidia-nvshmem-cu12
 
 # Install cudnn package based on CUDA version
 if [[ "$CUDA_VERSION" == *"cu13"* ]]; then

requirements.txt

@@ -0,0 +1,13 @@
+apache-tvm-ffi==0.1.0b15
+click
+einops
+ninja
+numpy
+nvidia-cudnn-frontend>=1.13.0
+nvidia-cutlass-dsl>=4.2.1
+nvidia-ml-py
+packaging>=24.2
+requests
+tabulate
+torch
+tqdm

setup.py

@@ -41,21 +41,21 @@ def generate_build_meta() -> None:
 
 ext_modules: List[setuptools.Extension] = []
 cmdclass: Mapping[str, type[setuptools.Command]] = {}
-install_requires = [
-    "numpy",
-    "torch",
-    "ninja",
-    "requests",
-    "nvidia-ml-py",
-    "einops",
-    "click",
-    "tqdm",
-    "tabulate",
-    "apache-tvm-ffi==0.1.0b15",
-    "packaging>=24.2",
-    "nvidia-cudnn-frontend>=1.13.0",
-    "nvidia-cutlass-dsl>=4.2.1",
-]
+
+
+def get_install_requires() -> List[str]:
+    """Read install requirements from requirements.txt."""
+    requirements_file = root / "requirements.txt"
+    if not requirements_file.exists():
+        return []
+    return [
+        line.strip()
+        for line in requirements_file.read_text().splitlines()
+        if line.strip() and not line.strip().startswith("#")
+    ]
+
+
+install_requires = get_install_requires()
 generate_build_meta()