From 9558fcf89d7c77711c5e928f73a9d98db4650b81 Mon Sep 17 00:00:00 2001
From: Huanyu He <hhy@meta.com>
Date: Sat, 13 Jul 2024 16:37:34 -0700
Subject: [PATCH] enable unit test gpu (#2228)

Summary:
Pull Request resolved: https://github.com/pytorch/torchrec/pull/2228

# context

Differential Revision: D51095381
---
 .github/workflows/unittest_ci.yml | 34 +++++++++++++++----------------
 1 file changed, 17 insertions(+), 17 deletions(-)

diff --git a/.github/workflows/unittest_ci.yml b/.github/workflows/unittest_ci.yml
index edf7a9fc1..cf315933a 100644
--- a/.github/workflows/unittest_ci.yml
+++ b/.github/workflows/unittest_ci.yml
@@ -5,18 +5,18 @@ name: Unit Test CI
 
 on:
   # TODO: re-enable when GPU unit tests are working
-  # push:
-  #   paths-ignore:
-  #     - "docs/*"
-  #     - "third_party/*"
-  #     - .gitignore
-  #     - "*.md"
-  # pull_request:
-  #   paths-ignore:
-  #     - "docs/*"
-  #     - "third_party/*"
-  #     - .gitignore
-  #     - "*.md"
+  push:
+    paths-ignore:
+      - "docs/*"
+      - "third_party/*"
+      - .gitignore
+      - "*.md"
+  pull_request:
+    paths-ignore:
+      - "docs/*"
+      - "third_party/*"
+      - .gitignore
+      - "*.md"
   workflow_dispatch:
 
 jobs:
@@ -29,19 +29,19 @@ jobs:
          - os: linux.2xlarge
           # ideally we run on 3.9 and 3.10 as well, however we are limited in resources.
            python-version: 3.8
-           python-tag: "py38"
-           cuda-tag: "cu11"
+           python-tag: "py39"
+           cuda-tag: "cu12"
     steps:
     # Checkout the repository to the GitHub Actions runner
     - name: Check ldd --version
       run: ldd --version
     - name: Checkout
-      uses: actions/checkout@v2
+      uses: actions/checkout@v3
     - name: Update pip
       run: |
         sudo yum update -y
         sudo yum -y install git python3-pip
-        sudo pip3 install --upgrade pip
+        # sudo pip3 install --upgrade pip
     - name: Setup conda
       run: |
         wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh
@@ -99,7 +99,7 @@ jobs:
       matrix:
         os: [linux.4xlarge.nvidia.gpu]
         python-version: [3.8]
-        cuda-tag: ["cu11"]
+        cuda-tag: ["cu12"]
     needs: build_on_cpu
     # the glibc version should match the version of the one we used to build the binary
     # for this case, it's 2.26