diff --git a/rocclr/runtime/device/gpu/gpudevice.cpp b/rocclr/runtime/device/gpu/gpudevice.cpp index 185866a765..931356480b 100644 --- a/rocclr/runtime/device/gpu/gpudevice.cpp +++ b/rocclr/runtime/device/gpu/gpudevice.cpp @@ -355,9 +355,15 @@ void NullDevice::fillDeviceInfo( // should be the size we can actually allocate at application // start. Note that it may not be a guarantee still as the // application progresses. +#if defined(BRAHMA) + info_.maxMemAllocSize_ = std::max( + cl_ulong(memInfo.cardMemAvailableBytes), + cl_ulong(memInfo.cardExtMemAvailableBytes)); +#else info_.maxMemAllocSize_ = std::max( cl_ulong(memInfo.cardLargestFreeBlockBytes), cl_ulong(memInfo.cardExtLargestFreeBlockBytes)); +#endif if (settings().apuSystem_) { info_.maxMemAllocSize_ = std::max(