From 8518263b80d2c512f475e11a2ba88091861a6c46 Mon Sep 17 00:00:00 2001
From: Frank Lee <somerlee.9@gmail.com>
Date: Tue, 7 Feb 2023 13:49:38 +0800
Subject: [PATCH] [test] fixed the triton version for testing (#2608)

---
 .github/workflows/build_on_pr.yml   | 1 +
 colossalai/amp/apex_amp/apex_amp.py | 1 +
 requirements/requirements-test.txt  | 2 +-
 3 files changed, 3 insertions(+), 1 deletion(-)

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