From 94a3d74e114135bca51dc1a85de1de107068116b Mon Sep 17 00:00:00 2001
From: Kevin Zimmerman <4733573+kczimm@users.noreply.github.com>
Date: Mon, 7 Aug 2023 19:16:27 -0500
Subject: [PATCH] remove clear_gpu_cache for use_lib feature

---
 pgml-extension/src/api.rs | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pgml-extension/src/api.rs b/pgml-extension/src/api.rs
index a556a33c3..d55d2897c 100644
--- 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) {