aparler/PyNorch
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

small fix threads per block sum

Browse source
This commit is contained in:
lucasdelimanogueira 2024-05-02 15:21:25 -03:00
parent 976f932007
commit a538ab4ecd
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
norch/csrc/cuda.cu
View file

@ -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) {