1 files changed, 4 insertions, 0 deletions
diff --git a/cpu_ref/rsCpuScript.cpp b/cpu_ref/rsCpuScript.cpp
index fb87dd1a..30b530b9 100644
--- a/cpu_ref/rsCpuScript.cpp
+++ b/cpu_ref/rsCpuScript.cpp
@@ -546,6 +546,10 @@ bool RsdCpuScriptImpl::forEachMtlsSetup(const Allocation ** ains,
         mtls->fep.dim.y = outType->getDimY();
         mtls->fep.dim.z = outType->getDimZ();
 
+    } else if (sc != nullptr) {
+        mtls->fep.dim.x = sc->xEnd;
+        mtls->fep.dim.y = sc->yEnd;
+        mtls->fep.dim.z = 0;
     } else {
         mCtx->getContext()->setError(RS_ERROR_BAD_SCRIPT,
                                      "rsForEach called with null allocations");