pytorch-pcen icon indicating copy to clipboard operation
pytorch-pcen copied to clipboard

Published 20 hours ago • verified publisher icon daemon

Question about pcen_cuda_kernel

Open chanil1218 opened this issue 6 years ago • 1 comments

Thanks you for sharing your code. When I'm reading your code, I've noticed that pcen_cuda_kernel is missing. Is that part not implemented yet or omitted?

chanil1218 avatar Jan 13 '19 15:01 chanil1218

Good question -- that's not implemented yet. Even though the existing code works on CUDA, the routine isn't optimized. I was planning to use cuPy and pynvrtc to implement a CUDA kernel for PCEN, similar to the fused QRNN kernel, but I haven't had the chance.

daemon avatar Jan 13 '19 22:01 daemon