Issue with CUDA kernel invocation (dim) fixed