Fix for Issue #60: RuntimeError: cudaMemcpy failed

src-d / kmcuda

Large scale K-means and K-nn implementation on NVIDIA GPU / CUDA

Other

797 stars 146 forks source link

Closed cstich closed 4 years ago

cstich commented 4 years ago

This is just a quick PR for the fix proposed by @bosmart in issue #60.

I haven't tested it but this fixes my memory allocation problem on my local machine.

vmarkovtsev commented 4 years ago

Thanks! It passed the CI, so I hope it is a correct fix.

cstich commented 4 years ago

Me too.