diff --git a/rocclr/device/gpu/gpusettings.cpp b/rocclr/device/gpu/gpusettings.cpp index 3602b1f9ab..219ee31a4d 100644 --- a/rocclr/device/gpu/gpusettings.cpp +++ b/rocclr/device/gpu/gpusettings.cpp @@ -294,7 +294,7 @@ bool Settings::create(const CALdeviceattribs& calAttr, bool reportAsOCL12Device, supportDepthsRGB_ = true; } if (use64BitPtr_) { - if ((GPU_ENABLE_LARGE_ALLOCATION) && (oclVersion_ == OpenCL20)) { + if (GPU_ENABLE_LARGE_ALLOCATION && (viPlus_ || (oclVersion_ == OpenCL20))) { maxAllocSize_ = 64ULL * Gi; } else { maxAllocSize_ = 4048 * Mi;