Searched defs:benchmarker_cls (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_benchmarking.py | 31 def get_counter_value(benchmarker_cls, fn_name): argument
50 def test_benchmark_smoke(self, benchmarker_cls, device): argument
65 def test_benchmark_cpu_smoke(self, benchmarker_cls, device="cpu"): argument
78 def test_benchmark_gpu_smoke(self, benchmarker_cls, device=GPU_TYPE): argument
93 self, benchmarker_cls, device="cpu" if HAS_CPU else GPU_TYPE argument
102 def test_benchmark_safely_infers_device_many_devices(self, benchmarker_cls): argument
|