From 0b11daa37b8dd147616a7236bad53530e21671e1 Mon Sep 17 00:00:00 2001
From: Charles Coulombe <ccoulombe@users.noreply.github.com>
Date: Wed, 15 Jan 2025 15:58:58 -0500
Subject: [PATCH 1/2] Update pyproject.toml

Added triton as dependency
---
 pyproject.toml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pyproject.toml b/pyproject.toml
index 2d3abcb8..5831fe66 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -17,6 +17,7 @@ classifiers = [
 ]
 dependencies = [
     "torch",
+    "triton",
     "ninja",
     "einops",
     "transformers",

From d85f6aa111678eacc494309e3a1c168bc8f5b2bd Mon Sep 17 00:00:00 2001
From: Charles Coulombe <ccoulombe@users.noreply.github.com>
Date: Thu, 16 Jan 2025 08:18:25 -0500
Subject: [PATCH 2/2] Use triton as dependency in setup.py

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 54c7589a..7c6196d7 100755
--- a/setup.py
+++ b/setup.py
@@ -371,7 +371,7 @@ def run(self):
         "packaging",
         "ninja",
         "einops",
-        # "triton",
+        "triton",
         "transformers",
         # "causal_conv1d>=1.4.0",
     ],