[Backend][ROCM] Add gfx1150 support. (#17508)
Enable support on ROCMTarget to compile gfx1150 and gfx1151.
---------
Signed-off-by: stanley-nod <stanley@nod-labs.com>
diff --git a/compiler/plugins/target/ROCM/ROCMTargetUtils.cpp b/compiler/plugins/target/ROCM/ROCMTargetUtils.cpp
index 38b739d..61fff22 100644
--- a/compiler/plugins/target/ROCM/ROCMTargetUtils.cpp
+++ b/compiler/plugins/target/ROCM/ROCMTargetUtils.cpp
@@ -173,7 +173,7 @@
if (chipArch < 6000)
return failure();
// Latest GFX arch supported is gfx115x.
- if (chipArch > 11000)
+ if (chipArch > 11500)
return failure();
// Get chip code from suffix. i.e gfx1103 -> `3`.
// gfx90a -> `a` == `10`.
diff --git a/compiler/src/iree/compiler/Codegen/Dialect/GPU/TargetUtils/KnownTargets.cpp b/compiler/src/iree/compiler/Codegen/Dialect/GPU/TargetUtils/KnownTargets.cpp
index 874a52a..e02e759 100644
--- a/compiler/src/iree/compiler/Codegen/Dialect/GPU/TargetUtils/KnownTargets.cpp
+++ b/compiler/src/iree/compiler/Codegen/Dialect/GPU/TargetUtils/KnownTargets.cpp
@@ -213,8 +213,8 @@
.Case("rx7800xt", TargetDetails{rdna3Wgp, &rx7800xtChip})
// https://www.techpowerup.com/gpu-specs/radeon-rx-7700-xt.c3911
.Case("rx7700xt", TargetDetails{rdna3Wgp, &rx7700xtChip})
- .Cases("rdna3", "gfx1100", "gfx1101", "gfx1102", "gfx1103",
- TargetDetails{rdna3Wgp, nullptr})
+ .Cases("rdna3", "gfx1100", "gfx1101", "gfx1102", "gfx1103", "gfx1150",
+ "gfx1151", TargetDetails{rdna3Wgp, nullptr})
.Default(std::nullopt);
}
