Fix coredump when cuda assert failed.
This commit is contained in:
parent
774a449c7b
commit
d4ba1d5f68
|
@ -425,7 +425,7 @@ void DynamicMemPoolBestFit::ReleaseDeviceRes() {
|
|||
auto &device_addr = iter->device_addr_base_;
|
||||
if (device_addr != nullptr) {
|
||||
if (!FreeDeviceMem(device_addr)) {
|
||||
MS_LOG(EXCEPTION) << "Free device memory[" << device_addr << "] error.";
|
||||
MS_LOG(ERROR) << "Free device memory[" << device_addr << "] error.";
|
||||
}
|
||||
device_addr = nullptr;
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue