From 3e736184311f096a8cb5104c7ff4f6e4ac303d2d Mon Sep 17 00:00:00 2001
From: Darren Hague <d.hague@sap.com>
Date: Mon, 25 Mar 2024 17:58:51 +0000
Subject: [PATCH] Fabric manager: support A100, H100, H200, B100 & B200 GPUs
 (#20)

---
 resources/install_fabricmanager.sh | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/resources/install_fabricmanager.sh b/resources/install_fabricmanager.sh
index 53798b3..729d824 100755
--- a/resources/install_fabricmanager.sh
+++ b/resources/install_fabricmanager.sh
@@ -6,7 +6,9 @@ source "$BIN_DIR"/set_env_vars.sh
 
 GPU_NAME=$("${NVIDIA_ROOT}"/bin/nvidia-smi -i 0 --query-gpu=name --format=csv,noheader)
 
-if [[ "$GPU_NAME" == *"A100"* ]]; then
+# Typical GPU name is something like "NVIDIA H100 80GB HBM3"
+# Fabric manager is required by the newer, bigger GPUs like A100, H100, etc. so we match those GPU types here
+if [[ "$GPU_NAME" =~ (A100|H100|H200|B100|B200) ]]; then
   OUTDIR=/out/nvidia-fabricmanager/$DRIVER_VERSION
   FABRICMANAGER_ARCHIVE="fabricmanager-linux-x86_64-$DRIVER_VERSION-archive"