xref: /aosp_15_r20/external/pytorch/aten/src/ATen/cudnn/README.md (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1All files living in this directory are written with the assumption that cuDNN is available,
2which means that these code are not guarded by `#if AT_CUDNN_ENABLED()`. Therefore, whenever
3you need to use definitions from here, please guard the `#include<ATen/cudnn/*.h>` and
4definition usages with `#if AT_CUDNN_ENABLED()` macro, e.g. [native/cudnn/BatchNorm.cpp](native/cudnn/BatchNorm.cpp).
5