Home
last modified time | relevance | path

Searched refs:device_lazy_init (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dmtia_extension.cpp149 torch::utils::device_lazy_init(at::kMTIA); in deviceCount()
154 torch::utils::device_lazy_init(at::kMTIA); in deviceSynchronize()
163 torch::utils::device_lazy_init(at::kMTIA); in exchangeDevice()
172 torch::utils::device_lazy_init(at::kMTIA); in maybeExchangeDevice()
182 torch::utils::device_lazy_init(at::kMTIA); in getDefaultStream()
188 torch::utils::device_lazy_init(at::kMTIA); in getCurrentStream()
194 torch::utils::device_lazy_init(at::kMTIA); in setCurrentStream()
200 torch::utils::device_lazy_init(at::kMTIA); in getCurrentDevice()
206 torch::utils::device_lazy_init(at::kMTIA); in setCurrentDevice()
/aosp_15_r20/external/pytorch/torch/csrc/mtia/
H A DModule.cpp54 torch::utils::device_lazy_init(at::kMTIA); in initModule()
59 torch::utils::device_lazy_init(at::kMTIA); in initModule()
65 torch::utils::device_lazy_init(at::kMTIA); in initModule()
70 torch::utils::device_lazy_init(at::kMTIA); in initModule()
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Ddevice_lazy_init.h26 void device_lazy_init(at::DeviceType device_type);
32 device_lazy_init(device.type()); in maybe_initialize_device()
H A Ddevice_lazy_init.cpp20 void device_lazy_init(at::DeviceType device_type) { in device_lazy_init() function
/aosp_15_r20/external/pytorch/torch/csrc/xpu/
H A DModule.cpp68 torch::utils::device_lazy_init(at::kXPU); in THXPModule_exchangeDevice_wrap()
85 torch::utils::device_lazy_init(at::kXPU); in THXPModule_maybeExchangeDevice_wrap()
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A DModule.cpp90 torch::utils::device_lazy_init(at::kCUDA); in THCPModule_setDevice_wrap()
105 torch::utils::device_lazy_init(at::kCUDA); in THCPModule_exchangeDevice()
120 torch::utils::device_lazy_init(at::kCUDA); in THCPModule_maybeExchangeDevice()
129 torch::utils::device_lazy_init(at::kCUDA); in THCPModule_getDevice_wrap() member in THCPModule_getDevice_wrap::torch::utils
156 torch::utils::device_lazy_init(at::kCUDA); in THCPModule_canDeviceAccessPeer_wrap()
/aosp_15_r20/external/pytorch/tools/autograd/templates/
H A Dpython_variable_methods.cpp499 torch::utils::device_lazy_init(at::kCUDA); in THPVariable_cuda()
522 torch::utils::device_lazy_init(at::kMTIA); in THPVariable_mtia()
545 torch::utils::device_lazy_init(at::kXPU); in THPVariable_xpu()
/aosp_15_r20/external/pytorch/
H A Dbuild_variables.bzl899 "torch/csrc/utils/device_lazy_init.cpp",