fix

esl-epfl · Sep 11, 2024 · a7e17bd · a7e17bd
1 parent e4ba80e
commit a7e17bd
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 7 deletions.
diff --git a/sw/applications/example_im2col/im2col_lib.h b/sw/applications/example_im2col/im2col_lib.h
@@ -44,7 +44,6 @@
 #define HW_CONFIG_SPC
 
 /* Defines which DMA channels are available to the SPC, depending on HW specifications */
-//@ToDo: see what include in the open source version
 #define SPC_CH_MASK 0b0001 
 
 /* Base address of the im2col SPC */
@@ -105,7 +104,7 @@
 #define OW_NHWC (FW * FH * CH * BATCH)
 #define OH_NHWC (N_PATCHES_W) * (N_PATCHES_H)
 
-#define START_ID 2
+#define START_ID 0
 
 #define TEST_EN 1
 

diff --git a/sw/applications/example_im2col/main.c b/sw/applications/example_im2col/main.c
@@ -53,7 +53,7 @@ int main()
     }
 
     /* Print the end word for verification */
-    printf("&\n\r");
+    PRINTF("&\n\r");
 
     return EXIT_SUCCESS;
 }
diff --git a/sw/device/lib/drivers/im2col_spc/im2col.c b/sw/device/lib/drivers/im2col_spc/im2col.c
@@ -26,14 +26,13 @@ int get_channel_number(uint32_t mask) {
     while (mask > 0) {
         channel++;
         if (mask & 1) {  // If the least significant bit is set
-            printf("%d\n", channel);
-            return channel;
+            return channel - 1;
         }
         mask >>= 1;  // Right shift the mask to check the next bit
     }
 
-    // If no bit is set, return 0 indicating no valid channel
-    return 0;
+    // If no bit is set, return -1 indicating no valid channel
+    return -1;
 }
 
 int run_im2col(im2col_trans_t trans){