[llvm] [TargetParser] Simplify getArchFamilyNameAMDGCN. NFC. (PR #186122)

[via llvm-commits]

llvmbot wrote:


<!--LLVM PR SUMMARY COMMENT-->

@llvm/pr-subscribers-backend-amdgpu

Author: Jay Foad (jayfoad)

<details>
<summary>Changes</summary>



---
Full diff: https://github.com/llvm/llvm-project/pull/186122.diff


1 Files Affected:

- (modified) llvm/lib/TargetParser/TargetParser.cpp (+5-16) 


``````````diff
diff --git a/llvm/lib/TargetParser/TargetParser.cpp b/llvm/lib/TargetParser/TargetParser.cpp
index 3dcd0e9990021..c5c2fb6b70e09 100644
--- a/llvm/lib/TargetParser/TargetParser.cpp
+++ b/llvm/lib/TargetParser/TargetParser.cpp
@@ -205,23 +205,12 @@ const GPUInfo *getArchEntry(AMDGPU::GPUKind AK, ArrayRef<GPUInfo> Table) {
 } // namespace
 
 StringRef llvm::AMDGPU::getArchFamilyNameAMDGCN(GPUKind AK) {
-  switch (AK) {
-  case AMDGPU::GK_GFX9_GENERIC:
-  case AMDGPU::GK_GFX9_4_GENERIC:
-    return "gfx9";
-  case AMDGPU::GK_GFX10_1_GENERIC:
-  case AMDGPU::GK_GFX10_3_GENERIC:
-    return "gfx10";
-  case AMDGPU::GK_GFX11_GENERIC:
-    return "gfx11";
-  case AMDGPU::GK_GFX12_GENERIC:
-  case AMDGPU::GK_GFX12_5_GENERIC:
-    return "gfx12";
-  default: {
-    StringRef ArchName = getArchNameAMDGCN(AK);
-    return ArchName.empty() ? "" : ArchName.drop_back(2);
-  }
+  StringRef ArchName = getArchNameAMDGCN(AK);
+  if (ArchName.ends_with("-generic")) {
+    // Return the part before the first '-', e.g. "gfx9-4-generic" -> "gfx9".
+    return ArchName.take_front(ArchName.find('-'));
   }
+  return ArchName.empty() ? "" : ArchName.drop_back(2);
 }
 
 StringRef llvm::AMDGPU::getArchNameAMDGCN(GPUKind AK) {

``````````

</details>


https://github.com/llvm/llvm-project/pull/186122