From 6896b4ea61ef9ca75c56059536258bb4b2afc6db Mon Sep 17 00:00:00 2001
From: Irina Demeshko <dirishka@gmail.com>
Date: Fri, 17 Nov 2023 10:40:21 -0800
Subject: [PATCH] Update src/cunumeric/unary/unary_red_template.inl

Co-authored-by: Manolis Papadakis <manopapad@gmail.com>
---
 src/cunumeric/unary/unary_red_template.inl | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/cunumeric/unary/unary_red_template.inl b/src/cunumeric/unary/unary_red_template.inl
index eecc45ddf..87750e17c 100644
--- a/src/cunumeric/unary/unary_red_template.inl
+++ b/src/cunumeric/unary/unary_red_template.inl
@@ -92,7 +92,11 @@ static void unary_red_template(TaskContext& context)
                     has_where ? inputs[1] : dummy_where,
                     scalars[0].value<int32_t>(),
                     scalars[1].value<UnaryRedCode>()};
-  op_dispatch(args.op_code, UnaryRedDispatch<KIND, false>{}, args);
+  if (has_where) {
+    op_dispatch(args.op_code, UnaryRedDispatch<KIND, true>{}, args);
+  } else {
+    op_dispatch(args.op_code, UnaryRedDispatch<KIND, false>{}, args);
+  }
 }
 
 }  // namespace cunumeric