Fix coredump when cuda assert failed.

This commit is contained in:
gaoyong10 2022-08-15 11:41:06 +08:00
parent 774a449c7b
commit d4ba1d5f68
1 changed files with 1 additions and 1 deletions

View File

@ -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;
}