Skip to content

Commit aa872a9

Browse files
committed
Set NCCL env profile
1 parent 46c2e09 commit aa872a9

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

.github/workflows/_test_maxtext_gke_xpk.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ jobs:
4848
ENVS: |
4949
CUDNN_FRONTEND_CUDART_LIB_NAME=libcudart.so.13;
5050
COMMAND: |
51-
source \${NCCL_LIB_DIR}/nccl-env-profile.sh;
51+
NCCL_LIB_DIR=\${NCCL_LIB_DIR} . \${NCCL_LIB_DIR}/nccl-env-profile.sh;
5252
nsys-jax --capture-range=cudaProfilerApi \
5353
--capture-range-end=stop \
5454
-o /opt/output/profile.zip \

0 commit comments

Comments
 (0)