Home
last modified time | relevance | path

Searched refs:GPUDeviceBenchmarkRequest (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/codegen/rocm/
H A Drocm_benchmark_request.py10 from torch._inductor.autotune_process import GPUDeviceBenchmarkRequest, TensorMeta
17 class ROCmBenchmarkRequest(GPUDeviceBenchmarkRequest):
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dautotune_process.py577 class GPUDeviceBenchmarkRequest(BenchmarkRequest): class
604 class TritonBenchmarkRequest(GPUDeviceBenchmarkRequest):
680 class CUDABenchmarkRequest(GPUDeviceBenchmarkRequest):