added docs for cublas support

utilityai · Feb 25, 2024 · f167755 · f167755
1 parent 20e607c
commit f167755
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/llama-cpp-2/src/lib.rs b/llama-cpp-2/src/lib.rs
@@ -8,6 +8,10 @@
 //! # Examples
 //!
 //! - [simple](https://github.com/utilityai/llama-cpp-rs/blob/main/llama-cpp-2/examples/simple.rs)
+//!
+//! # Feature Flags
+//!
+//! - `cublas` enables CUDA gpu support.
 use std::ffi::NulError;
 use std::fmt::Debug;
 use std::num::NonZeroI32;