transformer: add debug info of snRuntime on core ids

pulp-platform · Oct 20, 2023 · 8ca7a89 · 8ca7a89
1 parent d61883e
commit 8ca7a89
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/sw/apps/transformer/src/transformer.h b/sw/apps/transformer/src/transformer.h
@@ -909,11 +909,8 @@ 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);
-                    }
+                    dump_id(compute_id);
+
                     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();