Update setup.py

pytorch · Jan 20, 2025 · 18111c9 · 18111c9
1 parent 14e7a15
commit 18111c9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -294,6 +294,7 @@ def get_extensions():
     package_data={
         "torchao.kernel.configs": ["*.pkl"],
     },
+    install_requires=["torch"],
     ext_modules=get_extensions() if use_cpp != "0" else None,
     extras_require={"dev": read_requirements("dev-requirements.txt")},
     description="Package for applying ao techniques to GPU models",