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>
This commit is contained in:
Yuxian Qiu
2025-07-28 17:01:39 +08:00
committed by GitHub
parent 4b4e4f20dd
commit dd6ed14acb
2 changed files with 6 additions and 0 deletions

View File

@@ -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/**/*',