diff --git a/benchmark/opperf/nd_operations/array_manipulation_operators.py b/benchmark/opperf/nd_operations/array_manipulation_operators.py index 98b22d9f0152..8cf723afd31e 100644 --- a/benchmark/opperf/nd_operations/array_manipulation_operators.py +++ b/benchmark/opperf/nd_operations/array_manipulation_operators.py @@ -68,6 +68,8 @@ def run_rearrange_operators_benchmarks(ctx=mx.cpu(), dtype='float32', profiler=' Context to run benchmarks dtype: str, default 'float32' Precision to use for benchmarks + profiler: str, default 'native' + Type of Profiler to use (native/python) warmup: int, default 25 Number of times to run for warmup runs: int, default 100 @@ -96,6 +98,8 @@ def run_shape_operators_benchmarks(ctx=mx.cpu(), dtype='float32', profiler='nati Context to run benchmarks dtype: str, default 'float32' Precision to use for benchmarks + profiler: str, default 'native' + Type of Profiler to use (native/python) warmup: int, default 25 Number of times to run for warmup runs: int, default 100 @@ -124,6 +128,8 @@ def run_expanding_operators_benchmarks(ctx=mx.cpu(), dtype='float32', profiler=' Context to run benchmarks dtype: str, default 'float32' Precision to use for benchmarks + profiler: str, default 'native' + Type of Profiler to use (native/python) warmup: int, default 25 Number of times to run for warmup runs: int, default 100 @@ -152,6 +158,8 @@ def run_rounding_operators_benchmarks(ctx=mx.cpu(), dtype='float32', profiler='n Context to run benchmarks dtype: str, default 'float32' Precision to use for benchmarks + profiler: str, default 'native' + Type of Profiler to use (native/python) warmup: int, default 25 Number of times to run for warmup runs: int, default 100