[CI, Minor] Regroup Gen-DGRL CI with other libs (#1814)

pytorch · Jan 18, 2024 · 2ecb709 · 2ecb709
1 parent eb61100
commit 2ecb709
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 42 deletions.
diff --git a/.github/workflows/test-linux-gendgrl.yml b/.github/workflows/test-linux-gendgrl.yml
diff --git a/.github/workflows/test-linux-libs.yml b/.github/workflows/test-linux-libs.yml
@@ -94,6 +94,31 @@ jobs:
  bash .github/unittest/linux_libs/scripts_d4rl/run_test.sh
  bash .github/unittest/linux_libs/scripts_d4rl/post_process.sh
 
+ unittests-gendgrl:
+ strategy:
+ matrix:
+ python_version: ["3.9"]
+ cuda_arch_version: ["12.1"]
+ if: ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Data') }}
+ uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
+ with:
+ repository: pytorch/rl
+ runner: "linux.g5.4xlarge.nvidia.gpu"
+ docker-image: "nvidia/cudagl:11.4.0-base"
+ timeout: 120
+ script: |
+ set -euo pipefail
+ export PYTHON_VERSION="3.9"
+ export CU_VERSION="cu117"
+ export TAR_OPTIONS="--no-same-owner"
+ export UPLOAD_CHANNEL="nightly"
+ export TF_CPP_MIN_LOG_LEVEL=0
+
+ bash .github/unittest/linux_libs/scripts_gen-dgrl/setup_env.sh
+ bash .github/unittest/linux_libs/scripts_gen-dgrl/install.sh
+ bash .github/unittest/linux_libs/scripts_gen-dgrl/run_test.sh
+ bash .github/unittest/linux_libs/scripts_gen-dgrl/post_process.sh
+
  unittests-gym:
  strategy:
  matrix: