diff --git a/rocclr/platform/command.cpp b/rocclr/platform/command.cpp index db0cce08bf..1e15aebead 100644 --- a/rocclr/platform/command.cpp +++ b/rocclr/platform/command.cpp @@ -457,7 +457,7 @@ int32_t NDRangeKernelCommand::captureAndValidate() { int32_t error; uint64_t lclMemSize = kernel().getDeviceKernel(device)->workGroupInfo()->localMemSize_; - parameters_ = kernel().parameters().capture(device, lclMemSize, &error); + parameters_ = kernel().parameters().capture(device, sharedMemBytes_ + lclMemSize, &error); return error; }