transformer: fix source for ifmap of concat layer

pulp-platform · Oct 20, 2023 · 1ebf880 · 1ebf880
1 parent 8461c86
commit 1ebf880
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sw/apps/transformer/src/transformer.h b/sw/apps/transformer/src/transformer.h
@@ -859,7 +859,7 @@ static inline void transformer_layer_fp64(transformer_layer_fp64_t *const l) {
                 snrt_dma_txid_t txid_ifmap = 
                     snrt_dma_start_2d(
                         ifmap_lin2,                                      /* dst */
-                        l->O_lin2 + ifmap_offset,                        /* src */
+                        l->ifmap_lin2 + ifmap_offset,                    /* src */
                         l->positional_embeddings_fa * sizeof(double),    /* size */
                         l->positional_embeddings_fa * sizeof(double),    /* dst_stride */
                         l->positional_embeddings_fa * sizeof(double),    /* src_stride */