Home
last modified time | relevance | path

Searched refs:enable_cuda_sync_events (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/profiler/orchestration/
H A Dobserver.cpp21 bool enable_cuda_sync_events, in ExperimentalConfig() argument
27 enable_cuda_sync_events{enable_cuda_sync_events},
H A Dobserver.h59 bool enable_cuda_sync_events = false,
76 bool enable_cuda_sync_events; member
/aosp_15_r20/external/pytorch/torch/_C/
H A D_profiler.pyi62 enable_cuda_sync_events: bool = ...,
/aosp_15_r20/external/pytorch/torch/csrc/profiler/
H A Dkineto_shim.cpp254 if (config.enable_cuda_sync_events || get_cuda_sync_enabled()) { in prepareTrace()
/aosp_15_r20/external/pytorch/torch/csrc/profiler/python/
H A Dinit.cpp373 p.enable_cuda_sync_events, in initPythonBindings()
/aosp_15_r20/external/pytorch/test/profiler/
H A Dtest_profiler.py1371 trace_and_check(exp_config=_ExperimentalConfig(enable_cuda_sync_events=True))