diff --git a/.github/workflows/build_on_pr.yml b/.github/workflows/build_on_pr.yml index 82b671ace..c7882db6e 100644 --- a/.github/workflows/build_on_pr.yml +++ b/.github/workflows/build_on_pr.yml @@ -52,6 +52,7 @@ jobs: **/*.h **/*.cpp **/*.cu + **/*.txt - name: List changed files run: | diff --git a/colossalai/amp/apex_amp/apex_amp.py b/colossalai/amp/apex_amp/apex_amp.py index 69a4e348e..e6bdbe452 100644 --- a/colossalai/amp/apex_amp/apex_amp.py +++ b/colossalai/amp/apex_amp/apex_amp.py @@ -2,6 +2,7 @@ # -*- encoding: utf-8 -*- import torch.nn as nn + try: import apex.amp as apex_amp except ImportError: diff --git a/requirements/requirements-test.txt b/requirements/requirements-test.txt index 9ef0a682b..93055cd12 100644 --- a/requirements/requirements-test.txt +++ b/requirements/requirements-test.txt @@ -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