recomment the cuda preprocessor check

ravenscroftj · Aug 26, 2023 · a00de2a · a00de2a
1 parent 215a69b
commit a00de2a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/gptj.cpp b/src/gptj.cpp
@@ -566,7 +566,7 @@ bool GPTJModel::load_model(std::string fname) {
 
 
 
-    //#if defined(GGML_USE_CLBLAST) || defined(GGML_USE_CUBLAS)
+    #if defined(GGML_USE_CLBLAST) || defined(GGML_USE_CUBLAS)
 
     if(config.n_gpu_layers > 0){
         size_t vram_total = 0;
@@ -603,7 +603,7 @@ bool GPTJModel::load_model(std::string fname) {
         spdlog::info("{}: [GPU] total VRAM used: {} MB\n", __func__, vram_total / 1024 / 1024);
     }
 
-    //#endif // defined(GGML_USE_CLBLAST) || defined(GGML_USE_CUBLAS)
+    #endif // defined(GGML_USE_CLBLAST) || defined(GGML_USE_CUBLAS)
 
     return true;
 }