Templates/Reduce1D: now properly loading 2D scalars

triton-lang · Jan 1, 2016 · 6ac5e1f · 6ac5e1f
1 parent cdff884
commit 6ac5e1f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/lib/kernels/templates/reduce_1d.cpp b/lib/kernels/templates/reduce_1d.cpp
@@ -147,10 +147,10 @@ std::string reduce_1d::generate_impl(std::string const & suffix, expression_tree
   stream << "unsigned int gsize = " <<GlobalSize0(backend) << ";" << std::endl;
 
   process(stream, PARENT_NODE_TYPE, {{"array1", "#scalartype #namereg = #pointer[#start];"},
-                                     {"arrayn", "#pointer += #start;"},
-                                     {"array1n", "#pointer += #start;"},
-                                     {"arrayn1", "#pointer += #start;"}},
-                                    expressions, mapping);
+                                     {"array11", "#scalartype #namereg = #pointer[#start];"},
+                                      {"arrayn", "#pointer += #start;"},
+                                      {"arrayn1", "#pointer += #start;"},
+                                      {"array1n", "#pointer += #start;"}}, expressions, mapping);
 
   for (unsigned int k = 0; k < N; ++k)
   {