remove clear_gpu_cache for use_lib feature (#899)

kczimm · web-flow · commit 4181e5f73f51 · 2023-08-07T19:16:47.000-05:00
diff --git a/pgml-extension/src/api.rs b/pgml-extension/src/api.rs
@@ -602,6 +602,7 @@ pub fn embed_batch(
 /// ```sql
 /// SELECT pgml.clear_gpu_cache(memory_usage => 0.5);
 /// ```
+#[cfg(all(feature = "python", not(feature = "use_as_lib")))]
 #[pg_extern(immutable, parallel_safe, name = "clear_gpu_cache")]
 pub fn clear_gpu_cache(memory_usage: default!(Option<f32>, "NULL")) -> bool {
     match crate::bindings::transformers::clear_gpu_cache(memory_usage) {
