added a warning if submodule not checked out

utilityai · Feb 1, 2024 · 1df79ce · 1df79ce
1 parent 8554bd7
commit 1df79ce
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/llama-cpp-sys-2/build.rs b/llama-cpp-sys-2/build.rs
@@ -1,11 +1,16 @@
 use std::env;
 use std::path::PathBuf;
+use std::path::Path;
 
 fn main() {
     println!("cargo:rerun-if-changed=llama.cpp");
 
     let cublas_enabled = env::var("CARGO_FEATURE_CUBLAS").is_ok();
 
+    if !Path::new("llama.cpp/ggml.c").exists() {
+    	panic!("llama.cpp seems to not be populated, try running `git submodule update --init --recursive` to init.")
+    }
+
     let mut ggml = cc::Build::new();
     let mut ggml_cuda = if cublas_enabled { Some(cc::Build::new()) } else { None };
     let mut llama_cpp = cc::Build::new();