Home
last modified time | relevance | path

Searched defs:hasPrimaryContext (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDAFunctions.cpp176 bool (*hasPrimaryContext)(DeviceIndex) = dummyHasPrimaryContext; variable
184 bool hasPrimaryContext(DeviceIndex device_index) { in hasPrimaryContext() function
/aosp_15_r20/external/pytorch/aten/src/ATen/mps/
H A DMPSHooks.h54 bool hasPrimaryContext(DeviceIndex device_index) const override { in hasPrimaryContext() function
/aosp_15_r20/external/pytorch/aten/src/ATen/detail/
H A DPrivateUse1HooksInterface.h37 virtual bool hasPrimaryContext(DeviceIndex device_index) const override { in hasPrimaryContext() function
H A DHIPHooksInterface.h62 bool hasPrimaryContext(DeviceIndex device_index) const override { in hasPrimaryContext() function
H A DXPUHooksInterface.h69 bool hasPrimaryContext(DeviceIndex device_index) const override { in hasPrimaryContext() function
H A DMTIAHooksInterface.h57 bool hasPrimaryContext(DeviceIndex device_index) const override { in hasPrimaryContext() function
H A DMPSHooksInterface.h94 bool hasPrimaryContext(DeviceIndex device_index) const override { in hasPrimaryContext() function
H A DCUDAHooksInterface.h116 bool hasPrimaryContext(DeviceIndex device_index) const override { in hasPrimaryContext() function
/aosp_15_r20/external/pytorch/aten/src/ATen/xpu/detail/
H A DXPUHooks.cpp83 bool XPUHooks::hasPrimaryContext(DeviceIndex device_index) const { in hasPrimaryContext() function in at::xpu::detail::XPUHooks
/aosp_15_r20/external/pytorch/test/cpp_extensions/open_registration_extension/pytorch_openreg/csrc/
H A DOpenRegHooks.cpp21 bool hasPrimaryContext(c10::DeviceIndex device_index) const override { in hasPrimaryContext() function
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/detail/
H A DCUDAHooks.cpp248 bool CUDAHooks::hasPrimaryContext(DeviceIndex device_index) const { in hasPrimaryContext() function in at::cuda::detail::CUDAHooks