Add gfx950 (Mi350) GPU architecture support

- Add Mi350 to SUPPORTED_HIP_ARCHITECTURES in CMakeLists.txt
- Add GPU_ARCH_NUMBER_Mi350 mapping to gfx950
- Create parameters_Mi350.json parameter file
- Add gfx950 to gpu_architectures dictionary in smm_acc.py
- Add gfx950 GPU properties to gpu_properties.json (CDNA4, MI350X)
- Update installation documentation to include Mi350

Author: gsitaram

CMakeLists.txt

@@ -110,7 +110,7 @@ set(SUPPORTED_CUDA_ARCHITECTURES
     V100
     A100
     H100)
-set(SUPPORTED_HIP_ARCHITECTURES Mi50 Mi100 Mi250 Mi300)
+set(SUPPORTED_HIP_ARCHITECTURES Mi50 Mi100 Mi250 Mi300 Mi350)
 set(WITH_GPU
     $<IF:$<STREQUAL:${USE_ACCEL},"opencl">,"","P100">
     CACHE
@@ -272,6 +272,7 @@ if (USE_ACCEL MATCHES "cuda|hip")
   set(GPU_ARCH_NUMBER_Mi100 gfx908)
   set(GPU_ARCH_NUMBER_Mi250 gfx90a)
   set(GPU_ARCH_NUMBER_Mi300 gfx942)
+  set(GPU_ARCH_NUMBER_Mi350 gfx950)
 endif ()
 
 if (USE_ACCEL MATCHES "cuda")

docs/guide/2-user-guide/1-installation/index.md

@@ -72,7 +72,7 @@ make
 -DWITH_HIP_PROFILING=<OFF|ON>
 -DWITH_C_API=<ON|OFF>
 -DWITH_EXAMPLES=<ON|OFF>
--DWITH_GPU=<P100|K20X|K40|K80|V100|Mi50|Mi100|Mi250|Mi300>
+-DWITH_GPU=<P100|K20X|K40|K80|V100|Mi50|Mi100|Mi250|Mi300|Mi350>
 -DCMAKE_BUILD_TYPE=<Release|Debug|Coverage>
 -DBUILD_TESTING=<ON|OFF>
 -DTEST_MPI_RANKS=<2|auto|N>

src/acc/libsmm_acc/kernels/gpu_properties.json

@@ -211,18 +211,33 @@
         "Memory_Interface_(bits)": 8192
     },
     "gfx942" : {
-        "GPU_Architecture": "CDNA2",
-        "Product_Names": ["Instinct MI300"],
+        "GPU_Architecture": "CDNA3",
+        "Product_Names": ["Instinct MI300A"],
         "Threads_/_Warp" : 64,
         "Warps_/_Multiprocessor": 32,
         "Threads_/_Multiprocessor": 2048,
         "Thread_Blocks_/_Multiprocessor": 32,
         "Max_Thread_Block_Size": 1024,
         "Max_Shared_Memory_/_Block_(bytes)": 64000,
         "Shared_Memory_/_Multiprocessor_(bytes)": 64000,
-        "Compute_Units": 304,
-        "Stream_Multiprocessors": 304,
+        "Compute_Units": 228,
+        "Stream_Multiprocessors": 14592,
         "Memory_Size_(GB)": 128,
         "Memory_Interface_(bits)": 8192
+    },
+    "gfx950" : {
+        "GPU_Architecture": "CDNA4",
+        "Product_Names": ["Instinct MI350X"],
+        "Threads_/_Warp" : 64,
+        "Warps_/_Multiprocessor": 32,
+        "Threads_/_Multiprocessor": 2048,
+        "Thread_Blocks_/_Multiprocessor": 32,
+        "Max_Thread_Block_Size": 1024,
+        "Max_Shared_Memory_/_Block_(bytes)": 64000,
+        "Shared_Memory_/_Multiprocessor_(bytes)": 64000,
+        "Compute_Units": 256,
+        "Stream_Multiprocessors": 16384,
+        "Memory_Size_(GB)": 288,
+        "Memory_Interface_(bits)": 8192
     }
 }

src/acc/libsmm_acc/kernels/smm_acc.py

@@ -65,6 +65,7 @@
     "parameters_Mi100.json": "gfx908",
     "parameters_Mi250.json": "gfx90a",
     "parameters_Mi300.json": "gfx942",
+    "parameters_Mi350.json": "gfx950",
 }