Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_dynamo/backends/
H A Dcudagraphs.py123 def cudagraphs(dynamo_model, dynamo_inputs): function
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dconfig.py833 cudagraphs = os.environ.get("TORCHINDUCTOR_CUDAGRAPHS") == "1" variable in triton
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_aotdispatch.py6431 def post_compile(self, gm, inputs, cudagraphs): argument