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 @@ ...@@ -14,6 +14,7 @@
from superbench.benchmarks import DistributedImpl, DistributedBackend, BenchmarkRegistry, ReturnCode, Precision from superbench.benchmarks import DistributedImpl, DistributedBackend, BenchmarkRegistry, ReturnCode, Precision
from superbench.benchmarks.micro_benchmarks import MicroBenchmark from superbench.benchmarks.micro_benchmarks import MicroBenchmark
from superbench.benchmarks.context import Enum from superbench.benchmarks.context import Enum
from superbench.benchmarks.reducer import ReduceType
class ComputationKernelType(Enum): class ComputationKernelType(Enum):
...@@ -390,7 +391,7 @@ def _process_data(self, step_times): ...@@ -390,7 +391,7 @@ def _process_data(self, step_times):
Return: Return:
True if _process_data succeeds. 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 False
return True 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