Add gfx1150 support.

nod-ai · May 25, 2024 · 1b9a088 · 1b9a088
1 parent c814496
commit 1b9a088
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -213,7 +213,7 @@ std::optional<TargetDetails> 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);
 }