From a538ab4ecd93c2572478cb059ef2e837e873c6f9 Mon Sep 17 00:00:00 2001 From: lucasdelimanogueira Date: Thu, 2 May 2024 15:21:25 -0300 Subject: [PATCH] small fix threads per block sum --- norch/csrc/cuda.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/norch/csrc/cuda.cu b/norch/csrc/cuda.cu index bf413e3..071828f 100644 --- a/norch/csrc/cuda.cu +++ b/norch/csrc/cuda.cu @@ -5,7 +5,7 @@ #define THREADS_PER_BLOCK 128 #define TILE_SIZE 32 -#define SHMEM_SIZE THREADS_PER_BLOCK_SUM * sizeof(float) +#define SHMEM_SIZE THREADS_PER_BLOCK * sizeof(float) __host__ void cpu_to_cuda(Tensor* tensor) {