Run GPU test in various environment

pfnet · Jul 7, 2023 · 5e74b14 · 5e74b14
1 parent 6f0d686
commit 5e74b14
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 1 deletion.
diff --git a/.pfnci/config.pbtxt b/.pfnci/config.pbtxt
@@ -14,6 +14,6 @@ configs {
  time_limit: {
  seconds: 3600
  }
- command: "bash .pfnci/run_gpu_test.sh"
+ command: "bash .pfnci/run_gpu_test_all.sh"
  }
 }
diff --git a/.pfnci/run_gpu_test.sh b/.pfnci/run_gpu_test.sh
@@ -11,6 +11,7 @@ set -eux
 
 nvidia-smi
 
+export DEBIAN_FRONTEND=noninteractive
 apt-get update && apt-get install -y \
  python3 \
  python3-pip \

diff --git a/.pfnci/run_gpu_test_all.sh b/.pfnci/run_gpu_test_all.sh
@@ -0,0 +1,12 @@
+#!/bin/bash
+set -eux
+
+for LIBTORCH in cu113/1.12.1 cu117/1.13.0; do
+ for OS in ubuntu20.04 ubuntu22.04; do
+ IFS=/ read -r LIBTORCH_ARCH LIBTORCH_VERSION <<< $LIBTORCH
+ LIBTORCH_VERSION=$LIBTORCH_VERSION \
+ LIBTORCH_ARCH=$LIBTORCH_ARCH \
+ OS=$OS \
+ bash .pfnci/run_gpu_test.sh
+ done
+done