From 1b9a088dd9a90b576783e33546a4f6cd9ff84635 Mon Sep 17 00:00:00 2001 From: stanley-nod Date: Sat, 25 May 2024 00:47:41 -0700 Subject: [PATCH] Add gfx1150 support. --- compiler/plugins/target/ROCM/ROCMTargetUtils.cpp | 2 +- .../compiler/Codegen/Dialect/GPU/TargetUtils/KnownTargets.cpp | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/compiler/plugins/target/ROCM/ROCMTargetUtils.cpp b/compiler/plugins/target/ROCM/ROCMTargetUtils.cpp index 38b739d2eb17..61fff2231ebd 100644 --- a/compiler/plugins/target/ROCM/ROCMTargetUtils.cpp +++ b/compiler/plugins/target/ROCM/ROCMTargetUtils.cpp @@ -173,7 +173,7 @@ LogicalResult setHIPGlobals(Location loc, llvm::Module *module, 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 874a52abc353..6796f9560326 100644 --- a/compiler/src/iree/compiler/Codegen/Dialect/GPU/TargetUtils/KnownTargets.cpp +++ b/compiler/src/iree/compiler/Codegen/Dialect/GPU/TargetUtils/KnownTargets.cpp @@ -213,7 +213,7 @@ std::optional getAMDGPUTargetDetails(StringRef target) { .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", + .Cases("rdna3", "gfx1100", "gfx1101", "gfx1102", "gfx1103", "gfx1150", TargetDetails{rdna3Wgp, nullptr}) .Default(std::nullopt); }