SWDEV-336024 - Add a check for nullptr in HeapBuffer()

Change-Id: Ib8a3f2757c8600c4bd5bed417b230a60d4cf3d42
This commit is contained in:
German Andryeyev
2022-05-16 09:41:47 -04:00
förälder f41f227a33
incheckning 2f04eb90d3
+3 -1
Visa fil
@@ -1899,7 +1899,9 @@ class Device : public RuntimeObject {
Memory* P2PStage() const { return p2p_stage_; }
//! Returns heap buffer object for device allocator
device::Memory* HeapBuffer() const { return heap_buffer_->getDeviceMemory(*this); }
device::Memory* HeapBuffer() const {
return (heap_buffer_ != nullptr) ? heap_buffer_->getDeviceMemory(*this) : nullptr;
}
//! Does this device allow P2P access?
bool P2PAccessAllowed() const { return (p2p_access_devices_.size() > 0) ? true : false; }