From f05609ce69c1e1a7dd008c49cf435ef74df84b69 Mon Sep 17 00:00:00 2001
From: Vitaliy Chiley <6439018+vchiley@users.noreply.github.com>
Date: Thu, 11 Jan 2024 13:57:49 -0800
Subject: [PATCH] Updt triton pin (#89)

* Update setup.py

make torch pin more flexible

* Update setup.py
---
 setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 4a5a45ab..6f390d55 100644
--- 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",