Merge pull request #103 from utilityai/99-how-to-use-gpu

added docs for `cublas` support
utilityai · Feb 25, 2024 · 18cc21d · 18cc21d
2 parents 20e607c + f167755
commit 18cc21d
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;