Unpin gxx_linux-64 requirement (facebookresearch#3655)

Summary:
The TestPartitioning.TestPartitioningBigRange test case fails on gcc version 13.2. We can avoid this by requiring gcc version 11.2 where the test case works.

Pull Request resolved: facebookresearch#3655

Differential Revision: D59988036

Pulled By: gtwang01

Author: gtwang01

.circleci/config.yml

@@ -37,7 +37,7 @@ jobs:
       - run:
           name: Install env using main channel
           command: |
-            conda install -y -q python=3.11 cmake make swig mkl=2023 mkl-devel=2023 numpy scipy pytest gxx_linux-64=11.2 sysroot_linux-64
+            conda install -y -q python=3.11 cmake make swig mkl=2023 mkl-devel=2023 numpy scipy pytest gxx_linux-64 sysroot_linux-64
       - run:
           name: Build all targets
           no_output_timeout: 30m

.github/actions/build_cmake/action.yml

@@ -41,30 +41,28 @@ runs:
 
         # install base packages for X86_64
         if [ "${{ runner.arch }}" = "X64" ]; then
-          # TODO: unpin versions for gxx_linux-64 and sysroot_linux-64 and merge it with ARM64 below
-          if [ "${{ inputs.rocm }}" = "ON" ]; then
-            conda install -y -q -c conda-forge gxx_linux-64 sysroot_linux-64
-          else
-            conda install -y -q -c conda-forge gxx_linux-64=11.2 sysroot_linux-64=2.28
-          fi
+          # TODO: merge this with ARM64
+          conda install -y -q -c conda-forge gxx_linux-64 sysroot_linux-64
           conda install -y -q mkl=2023 mkl-devel=2023
         fi
 
-
         # install CUDA packages
         if [ "${{ inputs.gpu }}" = "ON" ] && [ "${{ inputs.raft }}" = "OFF" ]; then
-          conda install -y -q cuda-toolkit -c "nvidia/label/cuda-11.8.0"
+          conda install -y -q cuda-toolkit -c "nvidia/label/cuda-12.4.0"
+          conda install -y -q cuda-toolkit -c "nvidia/label/cuda-12.4.0"
         fi
 
         # install RAFT packages
         if [ "${{ inputs.raft }}" = "ON" ]; then
-          conda install -y -q libraft cuda-version=11.8 cuda-toolkit -c rapidsai-nightly -c "nvidia/label/cuda-11.8.0" -c conda-forge
+          conda install -y -q libraft cuda-version=12.4 cuda-toolkit -c rapidsai-nightly -c "nvidia/label/cuda-12.4.0" -c conda-forge
+          conda install -y -q libraft cuda-version=12.4 cuda-toolkit -c rapidsai-nightly -c "nvidia/label/cuda-12.4.0" -c conda-forge
         fi
 
         # install test packages
         conda install -y pytest
         if [ "${{ inputs.gpu }}" = "ON" ]; then
-          conda install -y -q pytorch pytorch-cuda=11.8 -c pytorch -c nvidia/label/cuda-11.8.0
+          conda install -y -q pytorch pytorch-cuda=12.4 -c pytorch -c nvidia/label/cuda-12.4.0
+          conda install -y -q pytorch pytorch-cuda=12.4 -c pytorch -c nvidia/label/cuda-12.4.0
         else
           conda install -y -q pytorch -c pytorch
         fi
@@ -118,6 +116,7 @@ runs:
       run: |
         eval "$(conda shell.bash hook)"
         conda activate
+
         cmake -B build \
               -DBUILD_TESTING=ON \
               -DBUILD_SHARED_LIBS=ON \
@@ -132,6 +131,9 @@ runs:
               -DCMAKE_CUDA_FLAGS=${{ runner.arch == 'X64' && '"-gencode arch=compute_75,code=sm_75"' || '' }} \
               .
         make -k -C build -j$(nproc)
+    - name: Setup tmate session
+      if: failure()
+      uses: mxschmitt/action-tmate@v3
     - name: C++ tests
       if: inputs.rocm == 'OFF'
       shell: bash

.github/workflows/build.yml

@@ -169,7 +169,6 @@ jobs:
         with:
           label: main
           cuda: "11.4.4"
-          compiler_version: "11.2"
   linux-x86_64-GPU-RAFT-packages-CUDA11-8-0:
     name: Linux x86_64 GPU w/ RAFT packages (CUDA 11.8.0)
     if: github.event_name == 'push' && startsWith(github.ref, 'refs/tags/v')
@@ -187,7 +186,6 @@ jobs:
           label: main
           raft: "ON"
           cuda: "11.8.0"
-          compiler_version: "11.2"
   linux-x86_64-GPU-packages-CUDA-12-1-1:
     name: Linux x86_64 GPU packages (CUDA 12.1.1)
     if: github.event_name == 'push' && startsWith(github.ref, 'refs/tags/v')
@@ -204,7 +202,6 @@ jobs:
         with:
           label: main
           cuda: "12.1.1"
-          compiler_version: "11.2"
   linux-x86_64-GPU-RAFT-packages-CUDA12-1-1:
     name: Linux x86_64 GPU w/ RAFT packages (CUDA 12.1.1)
     if: github.event_name == 'push' && startsWith(github.ref, 'refs/tags/v')
@@ -222,7 +219,6 @@ jobs:
           label: main
           raft: "ON"
           cuda: "12.1.1"
-          compiler_version: "11.2"
   windows-x86_64-packages:
     name: Windows x86_64 packages
     if: github.event_name == 'push' && startsWith(github.ref, 'refs/tags/v')