Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dconfig.py1127 cuda_backend = "triton" variable
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dcommon.py219 lambda *args, **kwargs: cuda_backends[config.cuda_backend](*args, **kwargs),