Searched refs:device_lazy_init (Results 1 – 8 of 8) sorted by relevance
149 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()
54 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()
26 void device_lazy_init(at::DeviceType device_type);32 device_lazy_init(device.type()); in maybe_initialize_device()
20 void device_lazy_init(at::DeviceType device_type) { in device_lazy_init() function
68 torch::utils::device_lazy_init(at::kXPU); in THXPModule_exchangeDevice_wrap()85 torch::utils::device_lazy_init(at::kXPU); in THXPModule_maybeExchangeDevice_wrap()
90 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::utils156 torch::utils::device_lazy_init(at::kCUDA); in THCPModule_canDeviceAccessPeer_wrap()
499 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()
899 "torch/csrc/utils/device_lazy_init.cpp",