24.04
|
Public Member Functions | |
def | __init__ (self, tol=0.01) |
def | add (self, BenchmarkResult benchmark_result) |
Generator[BenchmarkResult, None, None] | get_record (self) |
def | get_best_gemm_configs (self) |
def | get_best_gemm_configs_as_sequence (self) |
def | get_config_distributions (self) |
def | get_best_gemm_strategies (self) |
def | save_to_jsons (self, out_dir, only_best_config=True) |
def | summary (self, sum_level=SummaryLevel.Short) |
Static Public Attributes | |
SummaryLevel = Enum("SummaryLevel", ["Short", "Detailed"]) | |
A recorder that records and organises GEMM Benchmark results, and produces various reports on the record.
Definition at line 201 of file GemmTuner.py.
def __init__ | ( | self, | |
tol = 0.01 |
|||
) |
Initializer
Definition at line 207 of file GemmTuner.py.
References GEMMBenchmarkResultRecorder._strategies, GEMMBenchmarkResultRecorder._tol, and arm_compute::test::validation.set().
def add | ( | self, | |
BenchmarkResult | benchmark_result | ||
) |
Add a benchmark result to the record.
Definition at line 215 of file GemmTuner.py.
References GEMMBenchmarkResultRecorder._strategies.
Referenced by GEMMBenchmarkResultRecorder.get_config_distributions().
def get_best_gemm_configs | ( | self | ) |
Get the best GEMMConfig set per GEMMParam per Strategy
Definition at line 229 of file GemmTuner.py.
References GEMMBenchmarkResultRecorder._tol, and GEMMBenchmarkResultRecorder.get_record().
Referenced by GEMMBenchmarkResultRecorder.get_best_gemm_configs_as_sequence(), and GEMMBenchmarkResultRecorder.summary().
def get_best_gemm_configs_as_sequence | ( | self | ) |
Get the best GEMMConfig set per GEMMParam per Strategy, and flatten the result into a sequence of BenchmarkResults
Definition at line 255 of file GemmTuner.py.
References GEMMBenchmarkResultRecorder.get_best_gemm_configs().
Referenced by GEMMBenchmarkResultRecorder.get_config_distributions(), and GEMMBenchmarkResultRecorder.save_to_jsons().
def get_best_gemm_strategies | ( | self | ) |
Get the best Stratey per GEMMParam
Definition at line 280 of file GemmTuner.py.
References GEMMBenchmarkResultRecorder.get_record().
Referenced by GEMMBenchmarkResultRecorder.save_to_jsons().
def get_config_distributions | ( | self | ) |
Return GEMMConfigDistribution for each strategy
Definition at line 268 of file GemmTuner.py.
References GEMMBenchmarkResultRecorder.add(), and GEMMBenchmarkResultRecorder.get_best_gemm_configs_as_sequence().
Generator[BenchmarkResult, None, None] get_record | ( | self | ) |
Return an iterator that iterates over the record.
Definition at line 224 of file GemmTuner.py.
Referenced by GEMMBenchmarkResultRecorder.get_best_gemm_configs(), GEMMBenchmarkResultRecorder.get_best_gemm_strategies(), GEMMBenchmarkResultRecorder.save_to_jsons(), and GEMMBenchmarkResultRecorder.summary().
def save_to_jsons | ( | self, | |
out_dir, | |||
only_best_config = True |
|||
) |
Save records to an output directory of JSON files. The directory is organized such that each strategy gets its own JSON file. The directory also includes a JSON file to define the best strategy per GEMM Param.
Definition at line 303 of file GemmTuner.py.
References GEMMBenchmarkResultRecorder._strategies, GemmTuner.check_out_path(), GemmTuner.dump_json(), update_supported_ops.format, GEMMBenchmarkResultRecorder.get_best_gemm_configs_as_sequence(), GEMMBenchmarkResultRecorder.get_best_gemm_strategies(), GEMMBenchmarkResultRecorder.get_record(), and caffe_mnist_image_extractor.str.
def summary | ( | self, | |
sum_level = SummaryLevel.Short |
|||
) |
Return the summary string of the record
Definition at line 347 of file GemmTuner.py.
References GEMMBenchmarkResultRecorder.get_best_gemm_configs(), GEMMBenchmarkResultRecorder.get_record(), and arm_compute.join().
|
static |
Definition at line 205 of file GemmTuner.py.