@ptrblck Is aten/src/THCUNN/generic/SpatialConvolutionMM.cu still the CUDA convolution implementation that pytorch uses? When I clone the pytorch repository with git clone --recursive https://github.com/pytorch/pytorch that file isn’t even included.