gpu: jit: conv: add retype to ab swap fma layout

oneapi-src · Nov 15, 2023 · 09c87c7 · 09c87c7
1 parent db01d62
commit 09c87c7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/gpu/jit/conv/plan.cpp b/src/gpu/jit/conv/plan.cpp
@@ -1241,8 +1241,9 @@ struct fma_context_t {
                                     stride_t(new_inner_stride)));
                         }
                     }
-                    return layout_t(layout.type(), layout.ndims(),
-                            layout.offset(), blocks);
+                    return maybe_retype_layout_for_mad(is_a,
+                            layout_t(layout.type(), layout.ndims(),
+                                    layout.offset(), blocks));
                 }
             }
             // XXX: type and layout.type() may be different here when using mad