chenchenghao/DeepGEMM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add torch as build dependency. (#139)

Browse Source 
* 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
Download Patch File Download Diff File Expand all files Collapse all files
pyproject.toml
+3
View File
@@ -0,0 +1,3 @@
[build-system]
requires = ["torch>=2.1.0"]
build-backend = "setuptools.build_meta"
setup.py
+3
View File
@@ -78,6 +78,9 @@ if __name__ == '__main__':
name='deep_gemm', name='deep_gemm',
version='2.0.0' + revision, version='2.0.0' + revision,
packages=find_packages('.'), packages=find_packages('.'),
install_requires=[
'torch>=2.1.0',
],
package_data={ package_data={
'deep_gemm': [ 'deep_gemm': [
'include/deep_gemm/**/*', 'include/deep_gemm/**/*',