transformer: fix single cluster test

pulp-platform · Oct 20, 2023 · eef0700 · eef0700
1 parent c26cc21
commit eef0700
Showing 1 changed file with 6 additions and 4 deletions.
diff --git 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 {