21.05
|
Data Structures | |
class | BenchmarkResult |
class | GEMMBenchmarkResultRecorder |
class | GEMMConfigDistribution |
class | GEMMParam |
class | Measurement |
class | NativeGEMMConfig |
class | ReshapedGEMMConfig |
class | ReshapedOnlyRHSGEMMConfig |
Functions | |
Dict[str, str] | parse_benchmark_commandline (str commandline) |
Functions. More... | |
Generator[BenchmarkResult, None, None] | extract_benchmark_results (Dict json_results, measurement_method="avg") |
def | parse_json (dir_name) |
def | check_out_path (out_path) |
def | dump_json (out_path, dict) |
def | main (args) |
Main. More... | |
Variables | |
Strategy = Enum("Strategy", ["Native", "ReshapedOnlyRHS", "Reshaped"]) | |
Types. More... | |
GEMMConfigT | |
dictionary | GEMM_CONFIG_FACTORY |
dictionary | EXAMPLE_FILE_2_STRATEGY |
dictionary | GEMM_EXAMPLE_ARGS_FACTORY |
string | BENCHMARK_RESULT_JSON_EXTENSION = "gemmtuner_benchmark" |
parser = argparse.ArgumentParser(description="CL GEMM Tuner") | |
dest | |
metavar | |
action | |
type | |
help | |
required | |
default | |
args = parser.parse_args() | |
logging_level = logging.DEBUG if args.debug else logging.INFO | |
level | |
def GemmTuner.check_out_path | ( | out_path | ) |
Definition at line 566 of file GemmTuner.py.
References arm_compute::test::validation.input.
Referenced by GEMMBenchmarkResultRecorder.save_to_jsons().
def GemmTuner.dump_json | ( | out_path, | |
dict | |||
) |
Definition at line 582 of file GemmTuner.py.
Referenced by GEMMBenchmarkResultRecorder.save_to_jsons().
Generator[BenchmarkResult, None, None] GemmTuner.extract_benchmark_results | ( | Dict | json_results, |
measurement_method = "avg" |
|||
) |
Parse the benchmark result and extract relevant information, namely: GEMM param, Strategy, GEMM config, Measurements
Definition at line 488 of file GemmTuner.py.
References parse_benchmark_commandline().
Referenced by main().
def GemmTuner.main | ( | args | ) |
Main.
Definition at line 593 of file GemmTuner.py.
References extract_benchmark_results(), and parse_json().
Dict[str, str] GemmTuner.parse_benchmark_commandline | ( | str | commandline | ) |
Functions.
Parse the benchmark example command-line string into a dictionary of command-line arguments
Definition at line 468 of file GemmTuner.py.
References arm_compute::utils.map().
Referenced by extract_benchmark_results().
def GemmTuner.parse_json | ( | dir_name | ) |
Glob all benchmark result json files and parse them into json objects (dicts).
Definition at line 558 of file GemmTuner.py.
Referenced by main().
action |
Definition at line 646 of file GemmTuner.py.
args = parser.parse_args() |
Definition at line 679 of file GemmTuner.py.
Referenced by arm_conv::pooling.a64_fp32_nhwc_avg_3x3_s1_output2x2_depthfirst_impl(), arm_conv::pooling.a64_fp32_nhwc_max_2x2_s1_output2x2_depthfirst_impl(), arm_conv::pooling.a64_s8_nhwc_max_2x2_s1_output2x2_depthfirst_impl(), arm_conv::pooling.a64_u8_nhwc_max_2x2_s1_output2x2_depthfirst_impl(), Graph.add_node(), CommandLineParser.add_option(), CommandLineParser.add_positional_option(), arm_compute::test.apply(), arm_compute::test::framework.apply_impl(), CLSynthetizeFunction< CLGEMMMatrixMultiplyReshapedOnlyRHSKernel >.configure(), NESynthetizeFunction< K >.configure(), CLSynthetizeFunctionWithZeroConstantBorder< K, bordersize >.configure(), NESynthetizeFunctionWithZeroConstantBorder< K, bordersize >.configure(), CLSynthetizeFunctionInitOutputWithZeroAndWithZeroConstantBorder< K, bordersize >.configure(), NESynthetizeFunctionWithZeroConstantKernelBorder< K >.configure(), arm_compute::graph::backends.create_named_function(), arm_compute::graph::backends.create_named_memory_managed_function(), GemmImplementation< Top, Tret, OutputStage >.do_cycle_estimate(), GemmImplementation< Top, Tret, Nothing >.do_cycle_estimate(), GemmImplementation< Top, Tret, OutputStage >.do_instantiate(), GemmImplementation< Top, Tret, Nothing >.do_instantiate(), GemmImplementation< Top, Tret, OutputStage >.do_is_supported(), GemmImplementation< Top, Tret, Nothing >.do_is_supported(), GemmHybrid< strategy, To, Tr >.estimate_cycles(), GemmInterleavedPretransposed2d< strategy, To, Tr >.estimate_cycles(), GemmInterleaved< strategy, To, Tr, OutputStage, MergeStep, ForceThreadColumns >.estimate_cycles(), arm_conv::pooling.find_implementation(), arm_gemm.find_implementation(), arm_compute::utility.for_each(), arm_compute::detail.for_each_error(), arm_gemm.gemm(), GemmImplementation< Top, Tret, OutputStage >.GemmImplementation(), GemmImplementation< Top, Tret, Nothing >.GemmImplementation(), GemmInterleavedPretransposed2d< strategy, To, Tr >.GemmInterleavedPretransposed2d(), GemvBatched< To, Tr >.GemvBatched(), GemvPretransposed< strategy, To, Tr >.GemvPretransposed(), arm_gemm.get_compatible_kernels(), PoolingImplementation< TInput, TOutput, OutputStage >.get_cycle_estimate(), arm_gemm.get_gemm_method(), PoolingImplementation< TInput, TOutput, OutputStage >.get_instance(), PoolingImplementation< TInput, TOutput, OutputStage >.get_is_supported(), Logger.log(), arm_compute.operator<<(), arm_conv::pooling.pooling(), QuantizeWrapper< To, Tr, Tgemm >.QuantizeWrapper(), arm_compute::support::cpp11.snprintf(), arm_compute::support::cpp11.stof(), arm_compute::logging.string_with_format(), arm_compute.to_string(), CLSynthetizeFunction< CLGEMMMatrixMultiplyReshapedOnlyRHSKernel >.validate(), and NESynthetizeFunction< K >.validate().
string BENCHMARK_RESULT_JSON_EXTENSION = "gemmtuner_benchmark" |
Definition at line 461 of file GemmTuner.py.
default |
Definition at line 668 of file GemmTuner.py.
dest |
Definition at line 644 of file GemmTuner.py.
dictionary EXAMPLE_FILE_2_STRATEGY |
Definition at line 434 of file GemmTuner.py.
dictionary GEMM_CONFIG_FACTORY |
Definition at line 426 of file GemmTuner.py.
dictionary GEMM_EXAMPLE_ARGS_FACTORY |
Definition at line 450 of file GemmTuner.py.
GEMMConfigT |
Definition at line 189 of file GemmTuner.py.
help |
Definition at line 648 of file GemmTuner.py.
Referenced by Option.set_help().
level |
Definition at line 681 of file GemmTuner.py.
Referenced by arm_compute::test::framework.operator<<(), arm_compute::test::framework.operator>>(), and arm_compute::test::framework.to_string().
logging_level = logging.DEBUG if args.debug else logging.INFO |
Definition at line 680 of file GemmTuner.py.
metavar |
Definition at line 645 of file GemmTuner.py.
parser = argparse.ArgumentParser(description="CL GEMM Tuner") |
Definition at line 640 of file GemmTuner.py.
Referenced by CommonGemmExampleOptions.CommonGemmExampleOptions(), CommonGraphOptions.CommonGraphOptions(), CommonGraphValidateOptions.CommonGraphValidateOptions(), CommonOptions.CommonOptions(), GraphValidateExample< DepthwiseConvolutionLayer, DepthConvolutionOptions, DepthConvolutionVerifyAccessor >.do_setup(), main(), and arm_compute::utils.run_example().
required |
Definition at line 652 of file GemmTuner.py.
Referenced by arm_compute.adjust_down(), and arm_compute.adjust_up().
Strategy = Enum("Strategy", ["Native", "ReshapedOnlyRHS", "Reshaped"]) |
Types.
Definition at line 41 of file GemmTuner.py.
type |
Definition at line 647 of file GemmTuner.py.