Skip to content

Commit 3ddfc54

Browse files
committed
b200 test
1 parent 5b67fb0 commit 3ddfc54

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

.github/workflows/eval-gms8k.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ on:
4242
paths:
4343
- '.github/workflows/eval-gms8k.yml'
4444
- '.github/workflows/eval-tmpl.yml'
45-
- 'benchmarks/benchmark_lib.sh'
45+
- 'benchmarks/benchmark_lib.sh'
4646

4747
jobs:
4848
eval:

runners/launch_b200-nvd.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ docker run --rm --init --network host --name $server_name \
4242
-e HF_TOKEN -e HF_HUB_CACHE -e MODEL -e TP -e CONC -e MAX_MODEL_LEN -e ISL -e OSL -e PORT=$PORT -e EP_SIZE -e DP_ATTENTION \
4343
-e NCCL_GRAPH_REGISTER=0 \
4444
-e TORCH_CUDA_ARCH_LIST="10.0" -e CUDA_DEVICE_ORDER=PCI_BUS_ID -e CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7" \
45-
-e PYTHONPYCACHEPREFIX=/tmp/pycache/ -e RESULT_FILENAME -e RANDOM_RANGE_RATIO -e NUM_PROMPTS \
45+
-e PYTHONPYCACHEPREFIX=/tmp/pycache/ -e RESULT_FILENAME -e RANDOM_RANGE_RATIO -e NUM_PROMPTS -e -e RUN_EVAL \
4646
--entrypoint=/bin/bash \
4747
$(echo "$IMAGE" | sed 's/#/\//') \
4848
benchmarks/"${EXP_NAME%%_*}_${PRECISION}_b200${FRAMEWORK_SUFFIX}_docker.sh"

0 commit comments

Comments
 (0)