From 8d13dedad0f91eb85c33a083393a21346e5fd291 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Herv=C3=A9=20Guillemet?= <hg@apteryx.fr>
Date: Sun, 5 May 2024 10:18:24 +0200
Subject: [PATCH] Add swap for pytorch github workflow

---
 .github/actions/deploy-ubuntu/action.yml | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/.github/actions/deploy-ubuntu/action.yml b/.github/actions/deploy-ubuntu/action.yml
index cc762b6af8c..f900f79aa50 100644
--- a/.github/actions/deploy-ubuntu/action.yml
+++ b/.github/actions/deploy-ubuntu/action.yml
@@ -238,6 +238,13 @@ runs:
           $SUDO swapon /swapfile
         fi
 
+        if [[ "$CI_DEPLOY_MODULE" == "pytorch" ]]; then
+          $SUDO fallocate -l 4GB /swapfile
+          $SUDO chmod 600 /swapfile
+          $SUDO mkswap /swapfile
+          $SUDO swapon /swapfile
+        fi
+
         if [[ "$CI_DEPLOY_PLATFORM" == "linux-armhf" ]] && [[ "$CI_DEPLOY_MODULE" == "flycapture" ]]; then
           echo Installing FlyCapture amdhf
           python3 -m gdown.cli https://drive.google.com/uc?id=16NuUBs2MXQpVYqzDCEr9KdMng-6rHuDI