transformer: add debug info of snRuntime

pulp-platform · Oct 20, 2023 · d61883e · d61883e
1 parent 3f07167
commit d61883e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/sw/apps/transformer/src/transformer.h b/sw/apps/transformer/src/transformer.h
@@ -909,6 +909,11 @@ 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
+                    if (core_id == 0) {
+                        dump_idx(compute_id);
+                        dump_idx(cluster_id);
+                        dump_idx(num_cores);
+                    }
                     uint32_t row_offset = (B_r_lin2 / num_cores) * compute_id * l->positional_embeddings_fa;
                     uint32_t ofmap_offset = (B_r_lin2 / num_cores) * compute_id * B_c_lin2;
                     uint32_t start_gemm = snrt_mcycle();