Back to Pytorch

README

aten/src/ATen/cudnn/README.md

2.11.0392 B
Original Source

All files living in this directory are written with the assumption that cuDNN is available, which means that these code are not guarded by #if AT_CUDNN_ENABLED(). Therefore, whenever you need to use definitions from here, please guard the #include<ATen/cudnn/*.h> and definition usages with #if AT_CUDNN_ENABLED() macro, e.g. native/cudnn/BatchNorm.cpp.