From dd6ed14acbc7445dcef224248a77ab4d22b5f240 Mon Sep 17 00:00:00 2001
From: Yuxian Qiu <142763828+yuxianq@users.noreply.github.com>
Date: Mon, 28 Jul 2025 17:01:39 +0800
Subject: [PATCH] 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>
---
 pyproject.toml | 3 +++
 setup.py       | 3 +++
 2 files changed, 6 insertions(+)
 create mode 100644 pyproject.toml

diff --git a/pyproject.toml b/pyproject.toml
new file mode 100644
index 0000000..b61abcd
--- /dev/null
+++ b/pyproject.toml
@@ -0,0 +1,3 @@
+[build-system]
+requires = ["torch>=2.1.0"]
+build-backend = "setuptools.build_meta"
diff --git a/setup.py b/setup.py
index 2f4a427..916f760 100644
--- 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/**/*',