Home
last modified time | relevance | path

Searched refs:kInDevice (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/
Ddevice_address.h59 enum class DeviceAddressStatus { kInDevice, kInHost, kInDeviceToHost, kInHostToDevice }; enumerator
83 virtual DeviceAddressStatus status() const { return DeviceAddressStatus::kInDevice; } in status()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_kernel_runtime.cc693 device_address->set_status(DeviceAddressStatus::kInDevice); in ClearKernelOutputAddress()
959 device_address->set_status(DeviceAddressStatus::kInDevice); in AddMemorySwapTask()
1003 device_address_swap_in->set_status(DeviceAddressStatus::kInDevice); in UpdateHostSwapInQueue()
1008 case DeviceAddressStatus::kInDevice: in UpdateHostSwapInQueue()
1011 device_address->set_status(DeviceAddressStatus::kInDevice); in UpdateHostSwapInQueue()
1015 while (device_address->status() != DeviceAddressStatus::kInDevice) { in UpdateHostSwapInQueue()
1017 device_address_swap_in->set_status(DeviceAddressStatus::kInDevice); in UpdateHostSwapInQueue()
1310 device_address->set_status(DeviceAddressStatus::kInDevice); in FreeKernelDynamicRes()
1324 device_address->set_status(DeviceAddressStatus::kInDevice); in FreeKernelDynamicRes()
Dgpu_device_address.h58 DeviceAddressStatus status_{DeviceAddressStatus::kInDevice};