Merge pull request #30 from dakinggg/patch-1

Update pyproject.toml with tighter torch pin

Author: dakinggg

MANIFEST.in

@@ -1,5 +1,6 @@
 recursive-include csrc *.h
 recursive-include csrc *.cu
+recursive-include csrc *.cuh
 
 recursive-include third_party/cutlass/include *.h
 recursive-include third_party/cutlass/include *.hpp

pyproject.toml

@@ -1,4 +1,4 @@
 # build requirements
 [build-system]
-requires = ["setuptools < 70.0.0", "packaging >= 21.0.0", "torch>=2.4.0,<3.0"]
+requires = ["setuptools < 81.0.0", "packaging >= 21.0.0", "torch>=2.4.0,<2.7"]
 build-backend = "setuptools.build_meta"

setup.py

@@ -57,7 +57,7 @@
 
 setup(
     name="grouped_gemm",
-    version="0.1.6",
+    version="0.2.0",
     author="Trevor Gale",
     author_email="tgale@stanford.edu",
     description="Grouped GEMM",