Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
ColossalAI
Commits
8518263b
Unverified
Commit
8518263b
authored
Feb 07, 2023
by
Frank Lee
Committed by
GitHub
Feb 07, 2023
Browse files
[test] fixed the triton version for testing (#2608)
parent
aa7e9e47
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
3 additions
and
1 deletion
+3
-1
.github/workflows/build_on_pr.yml
.github/workflows/build_on_pr.yml
+1
-0
colossalai/amp/apex_amp/apex_amp.py
colossalai/amp/apex_amp/apex_amp.py
+1
-0
requirements/requirements-test.txt
requirements/requirements-test.txt
+1
-1
No files found.
.github/workflows/build_on_pr.yml
View file @
8518263b
...
@@ -52,6 +52,7 @@ jobs:
...
@@ -52,6 +52,7 @@ jobs:
**/*.h
**/*.h
**/*.cpp
**/*.cpp
**/*.cu
**/*.cu
**/*.txt
-
name
:
List changed files
-
name
:
List changed files
run
:
|
run
:
|
...
...
colossalai/amp/apex_amp/apex_amp.py
View file @
8518263b
...
@@ -2,6 +2,7 @@
...
@@ -2,6 +2,7 @@
# -*- encoding: utf-8 -*-
# -*- encoding: utf-8 -*-
import
torch.nn
as
nn
import
torch.nn
as
nn
try
:
try
:
import
apex.amp
as
apex_amp
import
apex.amp
as
apex_amp
except
ImportError
:
except
ImportError
:
...
...
requirements/requirements-test.txt
View file @
8518263b
...
@@ -9,5 +9,5 @@ torchaudio
...
@@ -9,5 +9,5 @@ torchaudio
torchrec==0.2.0
torchrec==0.2.0
contexttimer
contexttimer
einops
einops
triton==2.0.0.dev20221
011
triton==2.0.0.dev20221
202
git+https://github.com/HazyResearch/flash-attention.git@c422fee3776eb3ea24e011ef641fd5fbeb212623#egg=flash_attn
git+https://github.com/HazyResearch/flash-attention.git@c422fee3776eb3ea24e011ef641fd5fbeb212623#egg=flash_attn
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment