Writing your own CUDA kernel (Part 1)

Posted on mar. 02 octobre 2018 in Tutorial by Laura Domine
Tagged with cuda, tensorflow, pytorch

First part of a tutorial serie to write your own CUDA kernel and use it in Tensorflow or PyTorch.


Continue reading

Writing your own CUDA kernel (Part 2)

Posted on mar. 02 octobre 2018 in Tutorial by Laura Domine
Tagged with cuda, tensorflow, pytorch

Second part of a tutorial serie to write your own CUDA kernel and use it in Tensorflow or PyTorch.


Continue reading