Commit b5be51ed authored by carlushuang's avatar carlushuang
Browse files

throw exception if have empty pointer

parent 784ef1d3
...@@ -31,6 +31,10 @@ void DeviceMem::ToDevice(const void* p) const ...@@ -31,6 +31,10 @@ void DeviceMem::ToDevice(const void* p) const
hip_check_error( hip_check_error(
hipMemcpy(mpDeviceBuf, const_cast<void*>(p), mMemSize, hipMemcpyHostToDevice)); hipMemcpy(mpDeviceBuf, const_cast<void*>(p), mMemSize, hipMemcpyHostToDevice));
} }
else
{
throw std::runtime_error("ToDevice with an empty pointer");
}
} }
void DeviceMem::FromDevice(void* p) const void DeviceMem::FromDevice(void* p) const
...@@ -39,6 +43,10 @@ void DeviceMem::FromDevice(void* p) const ...@@ -39,6 +43,10 @@ void DeviceMem::FromDevice(void* p) const
{ {
hip_check_error(hipMemcpy(p, mpDeviceBuf, mMemSize, hipMemcpyDeviceToHost)); hip_check_error(hipMemcpy(p, mpDeviceBuf, mMemSize, hipMemcpyDeviceToHost));
} }
else
{
throw std::runtime_error("FromDevice with an empty pointer");
}
} }
void DeviceMem::SetZero() const void DeviceMem::SetZero() const
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment