Skip to content

Commit

Permalink
bump stk, gg
Browse files Browse the repository at this point in the history
  • Loading branch information
eitanturok authored Aug 19, 2024
1 parent 3a8cea0 commit 8d0349f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,13 +64,13 @@
'packaging>=21.3.0,<24.2',
'torch>=2.3.0,<2.5',
'triton>=2.1.0',
'stanford-stk @ git+https://git@github.com/stanford-futuredata/stk.git@a1ddf98466730b88a2988860a9d8000fd1833301',
'stanford-stk @ git+https://git@github.com/eitanturok/stk.git@bump-version',
]

extra_deps = {}

extra_deps['gg'] = [
'grouped_gemm @ git+https://git@github.com/tgale96/grouped_gemm.git@66c7195e35e8c4f22fa6a014037ef511bfa397cb',
'grouped_gemm @ git+https://git@github.com/eitanturok/grouped_gemm.git@bump-version',
]

extra_deps['dev'] = [
Expand Down

0 comments on commit 8d0349f

Please sign in to comment.