Merge pull request #4 from vchiley/updt_torch_pin

Updt triton pin
stanford-futuredata · Jan 11, 2024 · 7696c74 · 7696c74
2 parents aa534b6 + 5115ad6
commit 7696c74
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -11,7 +11,7 @@
 
 setup(
     name="stanford-stk",
-    version="0.0.6",
+    version="0.0.7",
     author="Trevor Gale",
     author_email="[email protected]",
     description="Sparse Toolkit",
@@ -25,7 +25,7 @@
     ],
     packages=find_packages(),
     install_requires=[
-        "triton==2.1.0",
+        "triton>=2.1.0",
     ],
     extras_require=extra_deps,
 )