Home
last modified time | relevance | path

Searched refs:deviceSynchronize (Results 1 – 16 of 16) sorted by relevance

/external/pytorch/torch/csrc/api/src/
Dmps.cpp24 at::detail::getMPSHooks().deviceSynchronize(); in synchronize()
Dxpu.cpp41 at::detail::getXPUHooks().deviceSynchronize( in synchronize()
Dcuda.cpp61 at::detail::getCUDAHooks().deviceSynchronize(device_index); in synchronize()
/external/pytorch/aten/src/ATen/xpu/detail/
DXPUHooks.h20 void deviceSynchronize(DeviceIndex device_index) const override;
DXPUHooks.cpp64 void XPUHooks::deviceSynchronize(DeviceIndex device_index) const { in deviceSynchronize() function in at::xpu::detail::XPUHooks
/external/pytorch/aten/src/ATen/detail/
DXPUHooksInterface.h57 virtual void deviceSynchronize(DeviceIndex /*device_index*/) const { in deviceSynchronize() function
DMTIAHooksInterface.h49 virtual void deviceSynchronize(c10::DeviceIndex device_index) const { in deviceSynchronize() function
DMPSHooksInterface.h40 virtual void deviceSynchronize() const { in deviceSynchronize() function
DCUDAHooksInterface.h195 virtual void deviceSynchronize(DeviceIndex /*device_index*/) const { in deviceSynchronize() function
/external/pytorch/aten/src/ATen/mps/
DMPSHooks.h25 void deviceSynchronize() const override;
DMPSHooks.mm66 void MPSHooks::deviceSynchronize() const {
/external/pytorch/aten/src/ATen/cuda/detail/
DCUDAHooks.h55 void deviceSynchronize(DeviceIndex device_index) const override;
DCUDAHooks.cpp457 void CUDAHooks::deviceSynchronize(DeviceIndex device_index) const { in deviceSynchronize() function in at::cuda::detail::CUDAHooks
/external/pytorch/torch/csrc/mtia/
DModule.cpp60 at::detail::getMTIAHooks().deviceSynchronize( in initModule()
/external/pytorch/torch/csrc/mps/
DModule.cpp81 at::detail::getMPSHooks().deviceSynchronize(); in MPSModule_deviceSynchronize()
/external/pytorch/test/cpp_extensions/
Dmtia_extension.cpp153 void deviceSynchronize(c10::DeviceIndex device_index) const override { in deviceSynchronize() function