"examples/industrial_data_pretraining/llm_asr/app.py" did not exist on "8c2527766639f8cd7a86400a154e0a2fe44c90e6"
rocm_onnxruntime_model_benchmark.py 762 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.

"""Docker benchmark example for onnxruntime models.

Commands to run:
  python3 examples/benchmarks/rocm_onnxruntime_model_benchmark.py
"""

from superbench.benchmarks import BenchmarkRegistry, Framework, Platform
from superbench.common.utils import logger

if __name__ == '__main__':
    context = BenchmarkRegistry.create_benchmark_context(
        'ort-models', platform=Platform.ROCM, framework=Framework.ONNXRUNTIME
    )

    benchmark = BenchmarkRegistry.launch_benchmark(context)
    if benchmark:
        logger.info(
            'benchmark: {}, return code: {}, result: {}'.format(
                benchmark.name, benchmark.return_code, benchmark.result
            )
        )