File tree Expand file tree Collapse file tree 1 file changed +2
-4
lines changed
script/app-mlperf-inference-mlcommons-python Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Original file line number Diff line number Diff line change @@ -499,7 +499,7 @@ def get_run_cmd_reference(
499
499
500
500
if env .get ('CM_ACTIVATE_RGAT_IN_MEMORY' , '' ) == "yes" :
501
501
cmd += " --in-memory "
502
-
502
+
503
503
elif "llama3" in env ['CM_MODEL' ]:
504
504
env ['RUN_DIR' ] = os .path .join (
505
505
env ['CM_MLPERF_INFERENCE_SOURCE' ],
@@ -520,12 +520,10 @@ def get_run_cmd_reference(
520
520
521
521
if env .get ('CM_MLPERF_INFERENCE_NUM_WORKERS' , '' ) != '' :
522
522
cmd += f" --num-workers { env ['CM_MLPERF_INFERENCE_NUM_WORKERS' ]} "
523
-
524
-
523
+
525
524
cmd = cmd .replace ("--count" , "--total-sample-count" )
526
525
cmd = cmd .replace ("--max-batchsize" , "--batch-size" )
527
526
528
-
529
527
if env .get ('CM_NETWORK_LOADGEN' , '' ) in ["lon" , "sut" ]:
530
528
cmd = cmd + " " + "--network " + env ['CM_NETWORK_LOADGEN' ]
531
529
if env .get ('CM_NETWORK_LOADGEN_SUT_SERVERS' , []):
You can’t perform that action at this time.
0 commit comments