Skip to content

Commit 4c50b2f

Browse files
authored
Update qa_l3_test.yml
Signed-off-by: zihugithub <103628180+zihugithub@users.noreply.github.com>
1 parent a49db40 commit 4c50b2f

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

.github/workflows/qa_l3_test.yml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,8 @@ jobs:
5555
NVTE_FRAMEWORK: pytorch
5656
TE_WITH_NCCL: 1
5757
run: |
58+
source /opt/miniconda3/etc/profile.d/conda.sh
59+
conda activate flagscale-train
5860
# Install MPI
5961
apt update
6062
apt install -y libopenmpi-dev openmpi-bin openmpi-common
@@ -94,6 +96,8 @@ jobs:
9496
TE_PATH: .
9597
MAX_JOBS: 32
9698
run: |
99+
source /opt/miniconda3/etc/profile.d/conda.sh
100+
conda activate flagscale-train
97101
mkdir -p "$XML_LOG_DIR"
98102
pip3 install pytest==8.2.1
99103
git clone https://github.com/Dao-AILab/flash-attention.git

0 commit comments

Comments
 (0)