Unverified Commit 4cb431ca authored by Ziyue Yang's avatar Ziyue Yang Committed by GitHub
Browse files

Benchmarks - Revise step time collection in distributed inference benchmark (#524)

**Description**
This commit revises distributed inference benchmark to give a unified
step time result by taking maximum step times of different GPUs.
parent 51761b3a
......@@ -14,6 +14,7 @@
from superbench.benchmarks import DistributedImpl, DistributedBackend, BenchmarkRegistry, ReturnCode, Precision
from superbench.benchmarks.micro_benchmarks import MicroBenchmark
from superbench.benchmarks.context import Enum
from superbench.benchmarks.reducer import ReduceType
class ComputationKernelType(Enum):
......@@ -390,7 +391,7 @@ def _process_data(self, step_times):
Return:
True if _process_data succeeds.
"""
if not self._process_numeric_result('step_times', step_times, cal_percentile=True):
if not self._process_numeric_result('step_times', step_times, reduce_type=ReduceType.MAX, cal_percentile=True):
return False
return True
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment