Repository: coreylowman/dfdx (language: rust)

Issue:
Add `cudnn` feature to feature flags documentation
Should say something like:
1. This requires "cuda" feature
2. This requires you install cuDNN (link to install page)


Discussion / hints:

Fix patch:
diff --git a/src/feature_flags.rs b/src/feature_flags.rs
index 53b0fa07c..ba93ab145 100644
--- a/src/feature_flags.rs
+++ b/src/feature_flags.rs
@@ -7,9 +7,9 @@
 //!
 //! # Quick start
 //!
-//! Cuda:
+//! Cuda (with cudnn):
 //! ```toml
-//! dfdx = { version = "...", default-features = false, features = ["std", "cuda"]}
+//! dfdx = { version = "...", default-features = false, features = ["std", "cuda", "cudnn"]}
 //! ```
 //!
 //! Cpu:
@@ -59,6 +59,17 @@
 //! dfdx = { version = "...", features = ["cuda"] }
 //! ```
 //!
+//! # "cudnn"
+//!
+//! This requires the "cuda" feature, and enables some cudnn optimizations.
+//! **This is purely for performance**, no other methods/structs are introduced
+//! with this feature.
+//!
+//! Example:
+//! ```toml
+//! dfdx = { version = "...", features = ["cudnn"] }
+//! ```
+//!
 //! # "cpu-seq-matmul"
 //!
 //! Used to enable using `matrixmultiply` for matmul operations.


Test patch: