Skip to content

Commit ebcca0e

Browse files
author
liyuzhuo
committed
wip
1 parent 6f68b6c commit ebcca0e

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

.github/workflows/qa-l0-te-cpp-unittest-pytorch-lint.yml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -174,10 +174,13 @@ jobs:
174174
# timeout-minutes: 10
175175
env:
176176
TE_PATH: .
177+
TE_FL_PREFER: vendor
177178
run: |
178179
# Activate conda environment
179180
source /opt/miniconda3/etc/profile.d/conda.sh
180181
conda activate flagscale-train
182+
183+
export TE_LIB_PATH=$(python -c "import site; print(site.getsitepackages()[0])")/transformer_engine
181184
182185
# Run core unit tests
183186
echo "=== Running L0 PyTorch Core Unit Tests ==="

0 commit comments

Comments
 (0)