From e14cadb1fb402aa37e277ded1139759b073453fd Mon Sep 17 00:00:00 2001 From: wozeparrot Date: Sun, 2 Aug 2026 21:52:36 +0800 Subject: [PATCH] gptoss: set ASM_GEMM (#17363) --- .../gpt_oss/implementations/tinybox_8xMI350X/dev_beam.sh | 1 + .../gpt_oss/implementations/tinybox_8xMI350X/dev_run.sh | 1 + 2 files changed, 2 insertions(+) diff --git a/examples/mlperf/training_submission_v6.1/tinycorp/benchmarks/gpt_oss/implementations/tinybox_8xMI350X/dev_beam.sh b/examples/mlperf/training_submission_v6.1/tinycorp/benchmarks/gpt_oss/implementations/tinybox_8xMI350X/dev_beam.sh index dfd115f544..35216e667b 100755 --- a/examples/mlperf/training_submission_v6.1/tinycorp/benchmarks/gpt_oss/implementations/tinybox_8xMI350X/dev_beam.sh +++ b/examples/mlperf/training_submission_v6.1/tinycorp/benchmarks/gpt_oss/implementations/tinybox_8xMI350X/dev_beam.sh @@ -10,6 +10,7 @@ export DEVICE_IN_FUNCTION_BUG=1 export DEBUG=${DEBUG:-2} export HK_FLASH_ATTENTION=${HK_FLASH_ATTENTION:-1} +export ASM_GEMM=${ASM_GEMM:-1} export ALL2ALL=${ALL2ALL:-1} export LATE_ALLREDUCE=${LATE_ALLREDUCE:-0} export ALLREDUCE_CAST=${ALLREDUCE_CAST:-1} diff --git a/examples/mlperf/training_submission_v6.1/tinycorp/benchmarks/gpt_oss/implementations/tinybox_8xMI350X/dev_run.sh b/examples/mlperf/training_submission_v6.1/tinycorp/benchmarks/gpt_oss/implementations/tinybox_8xMI350X/dev_run.sh index a8885d7605..9e071ea636 100755 --- a/examples/mlperf/training_submission_v6.1/tinycorp/benchmarks/gpt_oss/implementations/tinybox_8xMI350X/dev_run.sh +++ b/examples/mlperf/training_submission_v6.1/tinycorp/benchmarks/gpt_oss/implementations/tinybox_8xMI350X/dev_run.sh @@ -10,6 +10,7 @@ export DEVICE_IN_FUNCTION_BUG=1 export DEBUG=${DEBUG:-0} export HK_FLASH_ATTENTION=${HK_FLASH_ATTENTION:-1} +export ASM_GEMM=${ASM_GEMM:-1} export ALL2ALL=${ALL2ALL:-1} export LATE_ALLREDUCE=${LATE_ALLREDUCE:-0} export ALLREDUCE_CAST=${ALLREDUCE_CAST:-1}