Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDAFunctions.h63 SyncDebugMode get_sync_debug_mode() { in get_sync_debug_mode() function
84 warning_state().get_sync_debug_mode() != SyncDebugMode::L_DISABLED)) { in memcpy_and_sync()
102 warning_state().get_sync_debug_mode() != SyncDebugMode::L_DISABLED)) { in stream_synchronize()
H A DCUDAFunctions.cpp147 if (warning_state().get_sync_debug_mode() == SyncDebugMode::L_ERROR) { in warn_or_error_on_sync()
149 } else if (warning_state().get_sync_debug_mode() == SyncDebugMode::L_WARN) { in warn_or_error_on_sync()
/aosp_15_r20/external/pytorch/docs/source/
H A Dcuda.rst25 get_sync_debug_mode
/aosp_15_r20/external/pytorch/torch/cuda/
H A D__init__.py1039 def get_sync_debug_mode() -> int: function
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A DModule.cpp916 auto debug_mode = c10::cuda::warning_state().get_sync_debug_mode(); in THCPModule_cudaGetSyncDebugMode()
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_utils.py1922 self.debug_mode_restore = torch.cuda.get_sync_debug_mode()