mirror of https://github.com/hpcaitech/ColossalAI
[test] fixed the triton version for testing (#2608)
parent
aa7e9e4794
commit
8518263b80
|
@ -52,6 +52,7 @@ jobs:
|
|||
**/*.h
|
||||
**/*.cpp
|
||||
**/*.cu
|
||||
**/*.txt
|
||||
|
||||
- name: List changed files
|
||||
run: |
|
||||
|
|
|
@ -2,6 +2,7 @@
|
|||
# -*- encoding: utf-8 -*-
|
||||
|
||||
import torch.nn as nn
|
||||
|
||||
try:
|
||||
import apex.amp as apex_amp
|
||||
except ImportError:
|
||||
|
|
|
@ -9,5 +9,5 @@ torchaudio
|
|||
torchrec==0.2.0
|
||||
contexttimer
|
||||
einops
|
||||
triton==2.0.0.dev20221011
|
||||
triton==2.0.0.dev20221202
|
||||
git+https://github.com/HazyResearch/flash-attention.git@c422fee3776eb3ea24e011ef641fd5fbeb212623#egg=flash_attn
|
||||
|
|
Loading…
Reference in New Issue