Home
last modified time | relevance | path

Searched refs:do_bench_using_profiling (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/test/inductor/
Dtest_inductor_utils.py9 from torch._inductor.utils import do_bench_using_profiling
29 res = do_bench_using_profiling(self._bench_fn)
Dtest_fp8.py370 compiled_latency = utils.do_bench_using_profiling(
373 eager_latency = utils.do_bench_using_profiling(
379 ln_latency = utils.do_bench_using_profiling(functools.partial(compiled_ln, x))
/external/pytorch/torch/_inductor/runtime/
Dtriton_heuristics.py671 from torch._inductor.utils import do_bench_using_profiling
673 return do_bench_using_profiling(kernel_call, warmup=10, rep=40)
/external/pytorch/torch/_inductor/
Dutils.py124 def do_bench_using_profiling(fn: Callable[[], Any], warmup=25, rep=100) -> float: function