Merge pull request #18 from eitanturok/bump-version

bump torch, gg version

pyproject.toml

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

setup.py

@@ -1,9 +1,13 @@
 import os
 from pathlib import Path
 
-import torch
 from setuptools import find_packages, setup
-from torch.utils.cpp_extension import BuildExtension, CUDAExtension
+
+try:
+    import torch
+    from torch.utils.cpp_extension import BuildExtension, CUDAExtension
+except ModuleNotFoundError as e:
+    raise ModuleNotFoundError("No module named 'torch'. `torch` is required to install `grouped_gemm`.",) from e
 
 if os.environ.get("TORCH_CUDA_ARCH_LIST"):
     # Let PyTorch builder to choose device to target for.
@@ -53,7 +57,7 @@
 
 setup(
     name="grouped_gemm",
-    version="0.1.5",
+    version="0.1.6",
     author="Trevor Gale",
     author_email="[email protected]",
     description="Grouped GEMM",