Home
last modified time | relevance | path

Searched refs:set_sync_debug_mode (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDAFunctions.h59 void set_sync_debug_mode(SyncDebugMode l) { in set_sync_debug_mode() function
/aosp_15_r20/external/pytorch/docs/source/
H A Dcuda.rst33 set_sync_debug_mode
/aosp_15_r20/external/pytorch/torch/cuda/
H A D__init__.py1012 def set_sync_debug_mode(debug_mode: Union[int, str]) -> None: function
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A DModule.cpp909 c10::cuda::warning_state().set_sync_debug_mode(l); in THCPModule_cudaSetSyncDebugMode()
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_utils.py1923 torch.cuda.set_sync_debug_mode(self.mode)
1926 torch.cuda.set_sync_debug_mode(self.debug_mode_restore)