Skip to content

Commit 21c91ab

Browse files
author
Jatin Gangani
committed
update regex 3
1 parent 7153d95 commit 21c91ab

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

.github/workflows/benchmark-multinode-tmpl.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -98,9 +98,9 @@ jobs:
9898
if [ -f "$result_file" ]; then
9999
echo "Processing $result_file"
100100
# Extract GPU count, prefill_gpus and decode_gpus from filename for tp_size calculation
101-
gpus=$(echo "$result_file" | sed -n "s/.*_gpus_\([0-9]*\).*\.json/\1/p")
102-
prefill_gpus=$(echo "$result_file" | sed -n "s/.*_ctx_\([0-9]*\).*\.json/\1/p")
103-
decode_gpus=$(echo "$result_file" | sed -n "s/.*_gen_\([0-9]*\).*\.json/\1/p")
101+
gpus=$(echo "$result_file" | sed -n "s/.*_gpus-\([0-9]*\).*\.json/\1/p")
102+
prefill_gpus=$(echo "$result_file" | sed -n "s/.*_ctx-\([0-9]*\).*\.json/\1/p")
103+
decode_gpus=$(echo "$result_file" | sed -n "s/.*_gen-\([0-9]*\).*\.json/\1/p")
104104
105105
echo "Extracted: gpus=$gpus, prefill_gpus=$prefill_gpus, decode_gpus=$decode_gpus"
106106
if [ -n "$gpus" ]; then

runners/launch_gb200-nv.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -284,7 +284,7 @@ if [[ $FRAMEWORK == "dynamo-trtllm" ]]; then
284284
echo "Processing concurrency $concurrency with $gpus GPUs (prefill_gpus=$prefill_gpus, decode_gpus=$decode_gpus): $result_file"
285285

286286
# Copy the result file to workspace with a unique name
287-
WORKSPACE_RESULT_FILE="$GITHUB_WORKSPACE/${RESULT_FILENAME}_${CONFIG_NAME}_conc${concurrency}_gpus${gpus}_ctx${prefill_gpus}_gen${decode_gpus}.json"
287+
WORKSPACE_RESULT_FILE="$GITHUB_WORKSPACE/${RESULT_FILENAME}_${CONFIG_NAME}_conc${concurrency}_gpus-${gpus}_ctx-${prefill_gpus}_gen-${decode_gpus}.json"
288288
cp "$result_file" "$WORKSPACE_RESULT_FILE"
289289

290290
echo "Copied result file to: $WORKSPACE_RESULT_FILE"

0 commit comments

Comments
 (0)