Skip to content

Commit c0cd731

Browse files
Use unique BASE_OUTPUT_PATH to prevent conflicts (GoogleCloudPlatform#766)
1 parent d383439 commit c0cd731

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

dags/sparsity_diffusion_devx/maxtext_moe_gpu_e2e.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,12 +39,14 @@ def run_maxtext_tests():
3939
"mixtral-8x7b-1node": (
4040
f"SCANNED_CHECKPOINT={SCANNED_CHECKPOINT} \
4141
UNSCANNED_CKPT_PATH={UNSCANNED_CKPT_PATH} \
42+
BASE_OUTPUT_PATH=$GCS_OUTPUT \
4243
bash end_to_end/gpu/mixtral/test_8x7b.sh",
4344
1,
4445
),
4546
"mixtral-8x7b-2node": (
4647
f"SCANNED_CHECKPOINT={SCANNED_CHECKPOINT} \
4748
UNSCANNED_CKPT_PATH={UNSCANNED_CKPT_PATH} \
49+
BASE_OUTPUT_PATH=$GCS_OUTPUT \
4850
bash end_to_end/gpu/mixtral/test_8x7b.sh",
4951
2,
5052
),

0 commit comments

Comments
 (0)