pytorch中_cudnn_convolution_full_forward的定义在哪里

Question

作为标题提到，我想找到_cudnn_convolution_full_forward的定义，但我搜索了所有项目 在pytorch并失败了。我无法找到并讲述这个功能。

任何人都可以帮助我？

Answer 1

这里定义了所有cudnn卷积函数： https://github.com/pytorch/pytorch/blob/1848cad10802db9fa0aa066d9de195958120d863/aten/src/ATen/native/cudnn/Conv.cpp

此函数在最新版本的pytorch中不再存在。最接近的是cudnn_convolution_forward。在版本0.1.12中，该函数位于同一文件中： https://github.com/pytorch/pytorch/blob/v0.1.12/torch/csrc/cudnn/Conv.cpp

我建议不要使用非公共API（一个以_开头）而使用公共方法，但你可能已经知道了。

换句话说，你应该使用

torch.backends.cudnn.enabled = True

然后conv2d或conv3d取决于您的使用情况。