rename class name

microsoft · Apr 11, 2022 · c6f6d4c · c6f6d4c
1 parent 3128e90
commit c6f6d4c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/superbench/benchmarks/docker_benchmarks/__init__.py b/superbench/benchmarks/docker_benchmarks/__init__.py
@@ -6,6 +6,6 @@
 from superbench.benchmarks.docker_benchmarks.docker_base import DockerBenchmark, CudaDockerBenchmark, \
     RocmDockerBenchmark
 from superbench.benchmarks.docker_benchmarks.rocm_onnxruntime_performance import RocmOnnxRuntimeModelBenchmark
-from superbench.benchmarks.docker_benchmarks.fambench import FAMBench
+from superbench.benchmarks.docker_benchmarks.fambench import FAMBenchBenchmark
 
-__all__ = ['DockerBenchmark', 'CudaDockerBenchmark', 'RocmDockerBenchmark', 'RocmOnnxRuntimeModelBenchmark', 'FAMBench']
+__all__ = ['DockerBenchmark', 'CudaDockerBenchmark', 'RocmDockerBenchmark', 'RocmOnnxRuntimeModelBenchmark', 'FAMBenchBenchmark']
diff --git a/superbench/benchmarks/docker_benchmarks/fambench.py b/superbench/benchmarks/docker_benchmarks/fambench.py
@@ -13,7 +13,7 @@
 from superbench.benchmarks.docker_benchmarks.docker_base import CudaDockerBenchmark
 
 
-class FAMBench(CudaDockerBenchmark):
+class FAMBenchBenchmark(CudaDockerBenchmark):
     """The FAMBench E2E model benchmark class."""
     def __init__(self, name, parameters=''):
         """Constructor.
@@ -79,4 +79,4 @@ def _process_raw_result(self, cmd_idx, raw_output):
         return True
 
 
-BenchmarkRegistry.register_benchmark('fambench', FAMBench, platform=Platform.CUDA)
+BenchmarkRegistry.register_benchmark('fambench', FAMBenchBenchmark, platform=Platform.CUDA)