Home
last modified time | relevance | path

Searched refs:time_with_torch_timer (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/benchmarks/dynamo/microbenchmarks/
H A Dinductor_cpu_atomic.py3 from benchmark_helper import time_with_torch_timer
40 time_with_torch_timer(torch_scatter_add, (dst, src, index)).mean * 1000
43 time_with_torch_timer(inductor_scatter_add, (dst, src, index)).mean * 1000
46 time_with_torch_timer(torch_scatter_add, (dst, src, worst_index)).mean
50 time_with_torch_timer(inductor_scatter_add, (dst, src, worst_index)).mean
H A Dinductor_mm.py2 from benchmark_helper import time_with_torch_timer
49 torch_ms = time_with_torch_timer(torch_mm, (a, b)).mean * 1000
51 triton_ms = time_with_torch_timer(triton_mm, (a, b)).mean * 1000
54 ind_aten_ms = time_with_torch_timer(inductor_aten_mm, (a, b)).mean * 1000
57 ind_triton_ms = time_with_torch_timer(inductor_triton_mm, (a, b)).mean * 1000
H A Dmatmul_relu.py1 from benchmark_helper import time_with_torch_timer
56 time_with_torch_timer(torch_mm, (a, b), string_id="torch mm")
57 time_with_torch_timer(torch_mm_relu, (a, b), string_id="torch mm + relu")
58 time_with_torch_timer(inductor_mm, (a, b), string_id="inductor mm")
H A Dinductor_bmm.py1 from benchmark_helper import time_with_torch_timer
37 torch_ms = time_with_torch_timer(torch_bmm, (a, b)).mean * 1000
40 ind_aten_ms = time_with_torch_timer(inductor_aten_bmm, (a, b)).mean * 1000
43 ind_triton_ms = time_with_torch_timer(inductor_triton_bmm, (a, b)).mean * 1000
H A Dbenchmark_helper.py4 def time_with_torch_timer(fn, args, kwargs=None, iters=100): function