test-windows-optdepts.yml

name: Unit-tests on Windows

on:
  pull_request:
  push:
    branches:
      - nightly
      - main
      - release/*
  workflow_dispatch:

concurrency:
  # Documentation suggests ${{ github.head_ref }}, but that's only available on pull_request/pull_request_target triggers, so using ${{ github.ref }}.
  # On master, we want all builds to complete even if merging happens faster to make it easier to discover at which point something broke.
  group: ${{ github.workflow }}-${{ github.ref == 'refs/heads/main' && format('ci-master-{0}', github.sha) || format('ci-{0}', github.ref) }}
  cancel-in-progress: true

permissions:
  id-token: write
  contents: read

jobs:
  unittests-cpu:
    strategy:
      matrix:
        python-version:
          - "3.10"
        runner: ["windows.4xlarge"]
        gpu-arch-type: ["cpu"]
      fail-fast: false
    uses: pytorch/test-infra/.github/workflows/windows_job.yml@main
    with:
      repository: pytorch/rl
      runner: ${{ matrix.runner }}
      timeout: 60
      test-infra-ref: main
      script: |
        set -euxo pipefail

        export PYTHON_VERSION=${{ matrix.python-version }}
        export CU_VERSION="cpu"
        export torch_cuda="False"

        # TODO: Port this to pytorch/test-infra/.github/workflows/windows_job.yml
        export PATH="/c/Jenkins/Miniconda3/Scripts:${PATH}"

        echo "PYTHON_VERSION: $PYTHON_VERSION"
        echo "CU_VERSION: $CU_VERSION"

        if [[ "${{ github.ref }}" =~ release/* ]]; then
          export RELEASE=1
          export TORCH_VERSION=stable
        else
          export RELEASE=0
          export TORCH_VERSION=nightly
        fi
        export TD_GET_DEFAULTS_TO_NONE=1

        ./.github/unittest/windows_optdepts/scripts/unittest.sh

#  unittests-gpu:
#    uses: pytorch/test-infra/.github/workflows/windows_job.yml@main
#    with:
#      runner: "windows.g5.4xlarge.nvidia.gpu"
#      repository: pytorch/rl
#      timeout: 40
#      script: |
#        set -euxo pipefail
#
#        export PYTHON_VERSION="3.9"
#        export CUDA_VERSION="11.6"
#        export CU_VERSION="cu116"
#        export torch_cuda="True"
#
#        # TODO: Port this to pytorch/test-infra/.github/workflows/windows_job.yml
#        export PATH="/c/Jenkins/Miniconda3/Scripts:${PATH}"
#
#        echo "PYTHON_VERSION: $PYTHON_VERSION"
#
#        if [[ "${{ github.ref }}" =~ release/* ]]; then
#          export RELEASE=1
#          export TORCH_VERSION=stable
#        else
#          export RELEASE=0
#          export TORCH_VERSION=nightly
#        fi
#
#        ## setup_env.sh
#        ./.github/unittest/windows_optdepts/scripts/setup_env.sh
#
#        ## Install CUDA
#        packaging/windows/internal/cuda_install.bat
#
#        ## Update CUDA Driver
#        packaging/windows/internal/driver_update.bat
#
#        ## install.sh
#        ./.github/unittest/windows_optdepts/scripts/install.sh
#
#        ## run_test.sh
#        ./.github/unittest/windows_optdepts/scripts/run_test.sh
#
#        ## post_process.sh
#        ./.github/unittest/windows_optdepts/scripts/post_process.sh