removed check on xnack_count from kokkos script

amd · Jan 23, 2025 · d4aa725 · d4aa725
1 parent f1c17d3
commit d4aa725
Showing 1 changed file with 1 addition and 11 deletions.
diff --git a/extras/scripts/kokkos_setup.sh b/extras/scripts/kokkos_setup.sh
@@ -6,7 +6,6 @@ MODULE_PATH=/etc/lmod/modules/misc/kokkos
 BUILD_KOKKOS=0
 ROCM_VERSION=6.0
 KOKKOS_ARCH_AMD_GFX942="OFF"
-KOKKOS_ARCH_AMD_GFX942_APU="OFF"
 KOKKOS_ARCH_AMD_GFX90A="OFF"
 KOKKOS_ARCH_VEGA90A="OFF"
 KOKKOS_VERSION="4.5.01"
@@ -110,15 +109,7 @@ else
       if [ "${AMDGPU_GFXMODEL}" = "gfx90a" ]; then
          KOKKOS_ARCH_AMD_GFX90A="ON"
       elif [ "${AMDGPU_GFXMODEL}" = "gfx942" ]; then
-	 # check if on APU
-	 XNACK_COUNT=`rocminfo | grep xnack | wc -l`
-         if [ ${XNACK_COUNT} -gt 1 ]; then
-	    # MI300X
-            KOKKOS_ARCH_AMD_GFX942="ON"
-         else
-	    # MI300A
-            KOKKOS_ARCH_AMD_GFX942_APU="ON"
-         fi
+         KOKKOS_ARCH_AMD_GFX942="ON"
       elif [ "${AMDGPU_GFXMODEL}" = "gfx900" ]; then
          KOKKOS_ARCH_VEGA90A="ON"
       fi
@@ -142,7 +133,6 @@ else
                  -DKokkos_ENABLE_HIP=ON \
 		 -DKokkos_ENABLE_OPENMP=ON \
                  -DKokkos_ARCH_AMD_GFX942=${KOKKOS_ARCH_AMD_GFX942} \
-                 -DKokkos_ARCH_AMD_GFX942_APU=${KOKKOS_ARCH_AMD_GFX942_APU} \
                  -DKokkos_ARCH_AMD_GFX90A=${KOKKOS_ARCH_AMD_GFX90A} \
                  -DKokkos_ARCH_VEGA90A=${KOKKOS_ARCH_VEGA90A} \
                  -DKokkos_ARCH_ZEN=ON \