Skip to content

Commit e0706e6

Browse files
authored
Merge pull request #1 from rkarhila-amd/update_AMD_dsr1fp4_image
Update amd dsr1fp4 images
2 parents a470d47 + 5bdbd31 commit e0706e6

2 files changed

Lines changed: 2 additions & 0 deletions

File tree

benchmarks/dsr1_fp4_mi355x_docker.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
# RESULT_FILENAME
1212
# NUM_PROMPTS
1313
export SGLANG_USE_AITER=1
14+
export ROCM_QUICK_REDUCE_QUANTIZATION=INT4
1415

1516
PREFILL_SIZE=196608
1617
if [[ "$ISL" == "8192" && "$OSL" == "1024" ]]; then

benchmarks/dsr1_fp4_mi355x_slurm.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
# RANDOM_RANGE_RATIO
1111
# RESULT_FILENAME
1212
export SGLANG_USE_AITER=1
13+
export ROCM_QUICK_REDUCE_QUANTIZATION=INT4
1314
SERVER_LOG=$(mktemp /tmp/server-XXXXXX.log)
1415

1516
PREFILL_SIZE=196608

0 commit comments

Comments
 (0)