Home
last modified time | relevance | path

Searched refs:FreeLaunchDeviceMem (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_launch_kernel.h38 void FreeLaunchDeviceMem() override = 0;
Dascend_bucket.cc90 launch_mul_->FreeLaunchDeviceMem(); in FreeAllDeviceMem()
94 launch_atomic_clean_->FreeLaunchDeviceMem(); in FreeAllDeviceMem()
Dascend_launch_mul.h40 void FreeLaunchDeviceMem() override;
Dascend_launch_mul.cc49 void AscendLaunchMul::FreeLaunchDeviceMem() { in FreeLaunchDeviceMem() function in mindspore::device::ascend::AscendLaunchMul
Dascend_launch_atomic_clean.h43 void FreeLaunchDeviceMem() override;
Dascend_launch_transdata.h49 void FreeLaunchDeviceMem() override;
Dascend_launch_atomic_clean.cc72 void AscendLaunchAtomicClean::FreeLaunchDeviceMem() { in FreeLaunchDeviceMem() function in mindspore::device::ascend::AscendLaunchAtomicClean
Dascend_launch_transdata.cc74 void AscendLaunchTransData::FreeLaunchDeviceMem() { in FreeLaunchDeviceMem() function in mindspore::device::ascend::AscendLaunchTransData
Dascend_device_address.cc286 launch_transdata_->FreeLaunchDeviceMem(); in SyncDeviceToHostAndConvertFormatBasedOnTransData()
300 launch_transdata_->FreeLaunchDeviceMem(); in SyncDeviceToHostAndConvertFormatBasedOnTransData()
304 launch_transdata_->FreeLaunchDeviceMem(); in SyncDeviceToHostAndConvertFormatBasedOnTransData()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_launch_kernel.h38 void FreeLaunchDeviceMem() override = 0;
Dgpu_launch_mul.h39 void FreeLaunchDeviceMem() override;
Dgpu_launch_mul.cc48 void GPULaunchMul::FreeLaunchDeviceMem() { in FreeLaunchDeviceMem() function in mindspore::device::gpu::GPULaunchMul
Dgpu_bucket.cc96 launch_mul_->FreeLaunchDeviceMem(); in FreeAllDeviceMem()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dlaunch_kernel.h42 virtual void FreeLaunchDeviceMem() = 0;