Home
last modified time | relevance | path

Searched defs:benchmark_gpu (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/runtime/
H A Dbenchmarking.py162 def benchmark_gpu(self: Self, *args: Any, **kwargs: Any) -> float: member in Benchmarker
180 def benchmark_gpu(self: Self, _callable: Callable[[], Any], **kwargs: Any) -> float: member in TritonBenchmarker