Skip to content

Commit c5ab5a8

Browse files
Update pyproject.toml
1 parent 7a0c1fb commit c5ab5a8

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pyproject.toml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ build-backend = "flit_core.buildapi"
44

55
[project]
66
name = "kron-torch"
7-
version = "0.2.2"
7+
version = "0.2.3"
88
description = "An implementation of PSGD Kron optimizer in PyTorch."
99
readme = { file = "README.md", content-type = "text/markdown" }
1010
license = { file = "LICENSE" }
@@ -34,6 +34,7 @@ classifiers = [
3434
dependencies = [
3535
"torch",
3636
"opt_einsum",
37+
"triton==3.0.0"
3738
]
3839

3940
[project.urls]

0 commit comments

Comments
 (0)