Skip to content

Commit 0cce0fa

Browse files
authored
Add triton as dependency (#675)
* Update pyproject.toml Added triton as dependency * Use triton as dependency in setup.py
1 parent 9182c93 commit 0cce0fa

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

pyproject.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ classifiers = [
1717
]
1818
dependencies = [
1919
"torch",
20+
"triton",
2021
"ninja",
2122
"einops",
2223
"transformers",

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -371,7 +371,7 @@ def run(self):
371371
"packaging",
372372
"ninja",
373373
"einops",
374-
# "triton",
374+
"triton",
375375
"transformers",
376376
# "causal_conv1d>=1.4.0",
377377
],

0 commit comments

Comments
 (0)