xe: jit: codegen: allow scalar-based reduce for incompatible layouts

oneapi-src · Aug 2, 2024 · 281dd3b · 281dd3b
1 parent ff680fc
commit 281dd3b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gpu/intel/jit/codegen/reduce.hpp b/src/gpu/intel/jit/codegen/reduce.hpp
@@ -135,9 +135,9 @@ class reduce_impl_t {
                 a = layout_t(a.type(), a.ndims(), 0, a_blocks);
                 return find_1d_tile(a, b);
             }
+            return tensor_t(std::vector<dim_t>(b.ndims(), 1));
         }
 
-        ir_assert(ok) << "Incompatible layouts for reduction.";
         ir_assert(dim_t(b0.stride) == 1)
                 << "Reduction is not supported for non-unit dst stride.";