transformer: test offsets in 2nd cluster and compute_id

pulp-platform · Oct 20, 2023 · 84a39a0 · 84a39a0
1 parent 2f3bc31
commit 84a39a0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/sw/apps/transformer/src/transformer.h b/sw/apps/transformer/src/transformer.h
@@ -912,6 +912,7 @@ static inline void transformer_layer_fp64(transformer_layer_fp64_t *const l) {
                 if (snrt_is_compute_core()) {
                     // compute the gemm for the current row block and column block
                     uint32_t row_offset = (B_r_lin2 / num_cores) * compute_id * l->positional_embeddings_fa;
+                    dump_id(compute_id);
                     dump_ct(row_offset);
                     uint32_t ofmap_offset = (B_r_lin2 / num_cores) * compute_id * B_c_lin2;
                     uint32_t start_gemm = snrt_mcycle();