[Core][AMD] Add AMD Instinct MI350 and MI355 products (ray-project#55853)

Signed-off-by: root <[email protected]>
Signed-off-by: Lehui Liu <[email protected]>

Author: vickytsang

python/ray/_private/accelerators/amd_gpu.py

@@ -21,6 +21,8 @@
     "0x74a2": "AMD-Instinct-MI308X-OAM",
     "0x74a9": "AMD-Instinct-MI300X-HF",
     "0x74a5": "AMD-Instinct-MI325X-OAM",
+    "0x75a0": "AMD-Instinct-MI350X-OAM",
+    "0x75a3": "AMD-Instinct-MI355X-OAM",
     "0x6798": "AMD-Radeon-R9-200-HD-7900",
     "0x6799": "AMD-Radeon-HD-7900",
     "0x679A": "AMD-Radeon-HD-7900",

python/ray/util/accelerators/accelerators.py

@@ -23,6 +23,8 @@
 AMD_INSTINCT_MI300x_HF = "AMD-Instinct-MI300X-HF"
 AMD_INSTINCT_MI308x = "AMD-Instinct-MI308X"
 AMD_INSTINCT_MI325x = "AMD-Instinct-MI325X-OAM"
+AMD_INSTINCT_MI350x = "AMD-Instinct-MI350X-OAM"
+AMD_INSTINCT_MI355x = "AMD-Instinct-MI355X-OAM"
 AMD_RADEON_R9_200_HD_7900 = "AMD-Radeon-R9-200-HD-7900"
 AMD_RADEON_HD_7900 = "AMD-Radeon-HD-7900"
 AWS_NEURON_CORE = "aws-neuron-core"