Fixed Error in src

NicolaDean · NicolaDean · commit b979bd29d10b · 2023-03-30T22:44:26.000+02:00
diff --git a/alexnet_benchmark.py b/alexnet_benchmark.py
@@ -78,7 +78,7 @@ def pruning_model_random(model, px):
 
 N_CLASSES       = 10
 IMG_SIZE        = 227
-BATCH_SIZE      = 32
+BATCH_SIZE      = 1
 INPUT_CHANNELS  = 3
 PRUNING_PARAMETER = 0.90
 
diff --git a/lib/sparse_conv.so b/lib/sparse_conv.so
diff --git a/src/sparse_conv.cu b/src/sparse_conv.cu
@@ -264,7 +264,7 @@ __global__ void sconv_batch_base(const int *rowptr, const int *colidx, const Dty
 #define BLOCK_SIZE 256 // 4*4*32
 #define WARP_SIZE 32
 #define VLEN 32
-#define OC_BLOCK 4
+#define OC_BLOCK 1
 #define DIVIDE_INTO(x,y) ((x + y - 1)/y)
 #define MIN(x,y) ((x < y)? x : y)
 #define SHMEM_SIZE 1024
