From db771f4bca23714d97da6558cfd5932f393668ec Mon Sep 17 00:00:00 2001
From: Fangjun Kuang <csukuangfj@gmail.com>
Date: Fri, 3 Jun 2022 14:41:01 +0800
Subject: [PATCH] Download cudnn for windows.

---
 .../nightly-tests-ubuntu-conda-cpu.yml         |  8 ++++----
 .../nightly-tests-ubuntu-conda-cuda.yml        |  8 ++++----
 .../nightly-tests-windows-pip-cuda.yml         |  2 +-
 .github/workflows/run-tests-windows-cuda.yml   | 18 +++++++++++++++---
 4 files changed, 24 insertions(+), 12 deletions(-)

diff --git a/.github/workflows/nightly-tests-ubuntu-conda-cpu.yml b/.github/workflows/nightly-tests-ubuntu-conda-cpu.yml
index be802f5..7d47131 100644
--- a/.github/workflows/nightly-tests-ubuntu-conda-cpu.yml
+++ b/.github/workflows/nightly-tests-ubuntu-conda-cpu.yml
@@ -19,7 +19,7 @@ name: Nightly tests ubuntu conda cpu
 on:
   push:
     branches:
-      - nightly-test
+      - nightly-test-2
 
   schedule:
     # minute (0-59)
@@ -44,13 +44,13 @@ jobs:
         id: set-matrix
         run: |
           # outputting for debugging purposes
-          python scripts/github_actions/generate_build_matrix.py
-          MATRIX=$(python scripts/github_actions/generate_build_matrix.py)
+          python scripts/github_actions/generate_build_matrix.py --test-only-latest-torch
+          MATRIX=$(python scripts/github_actions/generate_build_matrix.py --test-only-latest-torch)
           echo "::set-output name=matrix::${MATRIX}"
 
   run_nightly_tests_ubuntu_conda_cpu:
     needs: generate_build_matrix
-    runs-on: ubuntu-18.04
+    runs-on: ubuntu-latest
     strategy:
       fail-fast: false
       matrix:
diff --git a/.github/workflows/nightly-tests-ubuntu-conda-cuda.yml b/.github/workflows/nightly-tests-ubuntu-conda-cuda.yml
index db1f450..521bbdd 100644
--- a/.github/workflows/nightly-tests-ubuntu-conda-cuda.yml
+++ b/.github/workflows/nightly-tests-ubuntu-conda-cuda.yml
@@ -19,7 +19,7 @@ name: Nightly tests ubuntu conda cuda
 on:
   push:
     branches:
-      - nightly-test
+      - nightly-test-2
 
   schedule:
     # minute (0-59)
@@ -44,13 +44,13 @@ jobs:
         id: set-matrix
         run: |
           # outputting for debugging purposes
-          python scripts/github_actions/generate_build_matrix.py --enable-cuda
-          MATRIX=$(python scripts/github_actions/generate_build_matrix.py --enable-cuda)
+          python scripts/github_actions/generate_build_matrix.py --enable-cuda --test-only-latest-torch
+          MATRIX=$(python scripts/github_actions/generate_build_matrix.py --enable-cuda --test-only-latest-torch)
           echo "::set-output name=matrix::${MATRIX}"
 
   run_nightly_tests_ubuntu_conda_cuda:
     needs: generate_build_matrix
-    runs-on: ubuntu-18.04
+    runs-on: ubuntu-latest
     strategy:
       fail-fast: false
       matrix:
diff --git a/.github/workflows/nightly-tests-windows-pip-cuda.yml b/.github/workflows/nightly-tests-windows-pip-cuda.yml
index f5684ea..8f4b27e 100644
--- a/.github/workflows/nightly-tests-windows-pip-cuda.yml
+++ b/.github/workflows/nightly-tests-windows-pip-cuda.yml
@@ -19,7 +19,7 @@ name: Nightly tests windows pip cuda
 on:
   push:
     branches:
-      - nightly-test
+      - nightly-test-2
 
   schedule:
     # minute (0-59)
diff --git a/.github/workflows/run-tests-windows-cuda.yml b/.github/workflows/run-tests-windows-cuda.yml
index 6f5b591..22edfba 100644
--- a/.github/workflows/run-tests-windows-cuda.yml
+++ b/.github/workflows/run-tests-windows-cuda.yml
@@ -41,8 +41,8 @@ jobs:
         id: set-matrix
         run: |
           # outputting for debugging purposes
-          python scripts/github_actions/generate_build_matrix.py --enable-cuda --for-windows
-          MATRIX=$(python scripts/github_actions/generate_build_matrix.py --enable-cuda --for-windows)
+          python scripts/github_actions/generate_build_matrix.py --enable-cuda --for-windows --test-only-latest-torch
+          MATRIX=$(python scripts/github_actions/generate_build_matrix.py --enable-cuda --for-windows --test-only-latest-torch)
           echo "::set-output name=matrix::${MATRIX}"
 
   run_tests_windows_cuda:
@@ -72,17 +72,29 @@ jobs:
 
       # See https://github.com/Jimver/cuda-toolkit/blob/master/src/links/windows-links.ts
       # for available CUDA versions
-      - uses: Jimver/cuda-toolkit@v0.2.5
+      - uses: Jimver/cuda-toolkit@v0.2.7
         id: cuda-toolkit
         with:
           cuda: ${{ matrix.cuda }}
 
       - name: Display CUDA version
+        shell: bash
         run: |
           echo "Installed cuda version is: ${{ steps.cuda-toolkit.outputs.cuda }}"
           echo "Cuda install location: ${{steps.cuda-toolkit.outputs.CUDA_PATH}}"
           nvcc --version
 
+      - name: Download cuDNN
+        shell: bash
+        run: |
+          GIT_LFS_SKIP_SMUDGE=1 git clone https://huggingface.co/csukuangfj/cudnn-for-windows
+          cd cudnn-for-windows
+          git lfs pull --include="cudnn-windows-x86_64-8.4.1.50_cuda11.6-archive.zip"
+          unzip cudnn-windows-x86_64-8.4.1.50_cuda11.6-archive.zip
+          ls -lh *
+          ls -lh */*
+
+
       - name: Install PyTorch ${{ matrix.torch }}
         shell: bash
         run: |