ci : enable test-grad0

ggerganov · Jul 6, 2023 · 1b9994f · 1b9994f
1 parent 2392f7a
commit 1b9994f
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 5 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -41,6 +41,10 @@ jobs:
   ubuntu-latest-cmake:
     runs-on: ubuntu-latest
 
+    env:
+      GGML_NLOOP: 3
+      GGML_NITER: 1
+
     steps:
       - name: Clone
         id: checkout
@@ -64,11 +68,15 @@ jobs:
         id: cmake_test
         run: |
           cd build
-          ctest --verbose
+          ctest --verbose --timeout 900
 
   ubuntu-latest-cmake-sanitizer:
     runs-on: ubuntu-latest
 
+    env:
+      GGML_NLOOP: 3
+      GGML_NITER: 1
+
     continue-on-error: true
 
     strategy:
@@ -99,7 +107,7 @@ jobs:
         id: cmake_test
         run: |
           cd build
-          ctest --verbose
+          ctest --verbose --timeout 900
 
   macOS-latest-make:
     runs-on: macos-latest
@@ -123,6 +131,10 @@ jobs:
   macOS-latest-cmake:
     runs-on: macos-latest
 
+    env:
+      GGML_NLOOP: 3
+      GGML_NITER: 1
+
     steps:
       - name: Clone
         id: checkout
@@ -147,10 +159,15 @@ jobs:
         id: cmake_test
         run: |
           cd build
-          ctest --verbose
+          ctest --verbose --timeout 900
 
   windows-latest-cmake:
     runs-on: windows-latest
+
+    env:
+      GGML_NLOOP: 3
+      GGML_NITER: 1
+
     env:
       OPENBLAS_VERSION: 0.3.23
       OPENCL_VERSION: 2023.04.17
@@ -249,7 +266,7 @@ jobs:
         if: ${{ matrix.build != 'clblast' && (matrix.build != 'avx512' || env.HAS_AVX512F == '1') }} # Test AVX-512 only when possible
         run: |
           cd build
-          ctest -C Release --verbose
+          ctest -C Release --verbose --timeout 900
 
       - name: Get commit hash
         id: commit

diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
@@ -10,5 +10,5 @@ llama_add_test(test-quantize-fns.cpp)
 llama_add_test(test-quantize-perf.cpp)
 llama_add_test(test-sampling.cpp)
 llama_add_test(test-tokenizer-0.cpp ${CMAKE_CURRENT_SOURCE_DIR}/../models/ggml-vocab.bin)
-# llama_add_test(test-grad0.c) # SLOW
+llama_add_test(test-grad0.c) # SLOW
 # llama_add_test(test-opt.c) # SLOW