From 13afb71208a985960fe8eec8eac0804158c54e85 Mon Sep 17 00:00:00 2001
From: Julien Chaumond <chaumond@gmail.com>
Date: Mon, 2 Mar 2020 11:56:45 -0500
Subject: [PATCH] [ci] Ensure that TF does not preempt all GPU memory for
 itself

see https://www.tensorflow.org/guide/gpu#limiting_gpu_memory_growth

Co-Authored-By: Funtowicz Morgan <mfuntowicz@users.noreply.github.com>
Co-Authored-By: Lysandre Debut <lysandre.debut@reseau.eseo.fr>
---
 .github/workflows/self-push.yml      | 1 +
 .github/workflows/self-scheduled.yml | 1 +
 2 files changed, 2 insertions(+)

diff --git a/.github/workflows/self-push.yml b/.github/workflows/self-push.yml
index cfbe999699..e21520d0e8 100644
--- a/.github/workflows/self-push.yml
+++ b/.github/workflows/self-push.yml
@@ -40,6 +40,7 @@ jobs:
 
     - name: Run all non-slow tests on GPU
       env:
+        TF_FORCE_GPU_ALLOW_GROWTH: yes
         OMP_NUM_THREADS: 1
         USE_CUDA: yes
       run: |
diff --git a/.github/workflows/self-scheduled.yml b/.github/workflows/self-scheduled.yml
index 7c33d5dfcb..d963ab5920 100644
--- a/.github/workflows/self-scheduled.yml
+++ b/.github/workflows/self-scheduled.yml
@@ -41,6 +41,7 @@ jobs:
 
     - name: Run all tests on GPU
       env:
+        TF_FORCE_GPU_ALLOW_GROWTH: yes
         OMP_NUM_THREADS: 1
         RUN_SLOW: yes
         USE_CUDA: yes