From 1b9994f8098b3bb49e82672ccec40a704769d07f Mon Sep 17 00:00:00 2001
From: Georgi Gerganov <ggerganov@gmail.com>
Date: Thu, 6 Jul 2023 20:57:12 +0300
Subject: [PATCH] ci : enable test-grad0

---
 .github/workflows/build.yml | 25 +++++++++++++++++++++----
 tests/CMakeLists.txt        |  2 +-
 2 files changed, 22 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 12481e8be..547b03a7a 100644
--- 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
index 4171c126c..1acf050a7 100644
--- 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