fix import

microsoft · Apr 18, 2023 · 8e138f4 · 8e138f4
1 parent e79352f
commit 8e138f4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/superbench/benchmarks/micro_benchmarks/dist_inference.py b/superbench/benchmarks/micro_benchmarks/dist_inference.py
@@ -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):