Updt triton pin (#89)

* Update setup.py make torch pin more flexible * Update setup.py
databricks · Jan 11, 2024 · f05609c · f05609c
1 parent 04e4f1f
commit f05609c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -29,7 +29,7 @@
 ]
 
 install_requires=[
-    "triton==2.1.0",
+    "triton>=2.1.0",
     "stanford-stk>=0.0.6",
 ]
 
@@ -51,7 +51,7 @@
 
 setup(
     name="megablocks",
-    version="0.5.0",
+    version="0.5.1",
     author="Trevor Gale",
     author_email="[email protected]",
     description="MegaBlocks",