Updt triton pin (#89)

vchiley · web-flow · commit f05609ce69c1 · 2024-01-11T16:57:49.000-05:00
* Update setup.py

make torch pin more flexible

* Update setup.py
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="tgale@stanford.edu",
     description="MegaBlocks",

Original file line number	Diff line number	Diff line change
`@@ -29,7 +29,7 @@`
`29`	`29`	`]`
`30`	`30`
`31`	`31`	`install_requires=[`
`32`		`- "triton==2.1.0",`
	`32`	`+ "triton>=2.1.0",`
`33`	`33`	`"stanford-stk>=0.0.6",`
`34`	`34`	`]`
`35`	`35`
`@@ -51,7 +51,7 @@`
`51`	`51`
`52`	`52`	`setup(`
`53`	`53`	`name="megablocks",`
`54`		`- version="0.5.0",`
	`54`	`+ version="0.5.1",`
`55`	`55`	`author="Trevor Gale",`
`56`	`56`	`author_email="tgale@stanford.edu",`
`57`	`57`	`description="MegaBlocks",`