From 1a34c3bca837e4f52211aa43c7c159dce9297274 Mon Sep 17 00:00:00 2001 From: Isuru Fernando Date: Mon, 2 Dec 2019 02:08:05 -0600 Subject: [PATCH] Don't crash if no device permission --- runtime/device/rocm/rocdevice.cpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/runtime/device/rocm/rocdevice.cpp b/runtime/device/rocm/rocdevice.cpp index df504f8..2eac260 100644 --- a/runtime/device/rocm/rocdevice.cpp +++ b/runtime/device/rocm/rocdevice.cpp @@ -765,7 +765,9 @@ bool Device::create(bool sramEccEnabled) { } } - xferQueue(); + if (xferQueue() == nullptr) { + return false; + } return true; } @@ -1810,6 +1812,7 @@ VirtualGPU* Device::xferQueue() const { thisDevice->xferQueue_ = reinterpret_cast(thisDevice->createVirtualDevice()); if (!xferQueue_) { LogError("Couldn't create the device transfer manager!"); + return nullptr; } } xferQueue_->enableSyncBlit();