diff --git a/sw/apps/transformer/src/transformer.h b/sw/apps/transformer/src/transformer.h index bbd438821..a99b5d4eb 100644 --- a/sw/apps/transformer/src/transformer.h +++ b/sw/apps/transformer/src/transformer.h @@ -922,10 +922,12 @@ static inline void transformer_layer_fp64(transformer_layer_fp64_t *const l) { snrt_cluster_hw_barrier(); - for (int i = 0; i < B_r_lin2 / num_cores; i++) { - for (int j = 0; j < B_c_lin2; j++) { - dump_idx(i * B_c_lin2 + j + ofmap_offset); - dump_debug(ofmap_lin2[i * B_c_lin2 + j + ofmap_offset]); + if (cluster_id == 0) { + for (int i = 0; i < B_r_lin2 / num_cores; i++) { + for (int j = 0; j < B_c_lin2; j++) { + dump_idx(i * B_c_lin2 + j + ofmap_offset); + dump_debug(ofmap_lin2[i * B_c_lin2 + j + ofmap_offset]); + } } } } else {