Add torch as build dependency. (#139)

* Add torch as dependency. Signed-off-by: Yuxian Qiu <142763828+yuxianq@users.noreply.github.com> * Add pyproject.toml Signed-off-by: Yuxian Qiu <142763828+yuxianq@users.noreply.github.com> * Fix setup.py. Signed-off-by: Yuxian Qiu <142763828+yuxianq@users.noreply.github.com> * Add build dependency. Signed-off-by: Yuxian Qiu <142763828+yuxianq@users.noreply.github.com> --------- Signed-off-by: Yuxian Qiu <142763828+yuxianq@users.noreply.github.com>
2025-07-28 17:01:39 +08:00
parent 4b4e4f20dd
commit dd6ed14acb
2 changed files with 6 additions and 0 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -0,0 +1,3 @@
+[build-system]
+requires = ["torch>=2.1.0"]
+build-backend = "setuptools.build_meta"
--- a/setup.py
+++ b/setup.py
@@ -78,6 +78,9 @@ if __name__ == '__main__':
        name='deep_gemm',
        version='2.0.0' + revision,
        packages=find_packages('.'),
+        install_requires=[
+            'torch>=2.1.0',
+        ],
        package_data={
            'deep_gemm': [
                'include/deep_gemm/**/*',