[AMDGPU] Whitelist all intrinsics (#130150)

For code maintainability -- this may result in cases where we are
applying the optimization where it is not profitable, but those are
likely to be rare.

Author: jrbyrnes

llvm/lib/Target/AMDGPU/AMDGPULateCodeGenPrepare.cpp

@@ -127,92 +127,8 @@ class LiveRegOptimizer {
     return LK.first != TargetLoweringBase::TypeLegal;
   }
 
-  /// Check if intrinsic natively operates on 8-bit or 16-bit
-  bool isNativeIntrinsic(Intrinsic::ID ID) {
-    switch (ID) {
-    case Intrinsic::amdgcn_dot4_f32_fp8_bf8:
-    case Intrinsic::amdgcn_dot4_f32_bf8_fp8:
-    case Intrinsic::amdgcn_dot4_f32_fp8_fp8:
-    case Intrinsic::amdgcn_dot4_f32_bf8_bf8:
-    case Intrinsic::amdgcn_mfma_i32_4x4x4i8:
-    case Intrinsic::amdgcn_mfma_i32_16x16x4i8:
-    case Intrinsic::amdgcn_mfma_i32_32x32x4i8:
-    case Intrinsic::amdgcn_mfma_i32_16x16x16i8:
-    case Intrinsic::amdgcn_mfma_i32_32x32x8i8:
-    case Intrinsic::amdgcn_mfma_i32_16x16x64_i8:
-    case Intrinsic::amdgcn_mfma_i32_32x32x32_i8:
-    case Intrinsic::amdgcn_mfma_i32_32x32x16_i8:
-    case Intrinsic::amdgcn_mfma_i32_16x16x32_i8:
-    case Intrinsic::amdgcn_mfma_f32_16x16x32_bf8_bf8:
-    case Intrinsic::amdgcn_mfma_f32_16x16x32_bf8_fp8:
-    case Intrinsic::amdgcn_mfma_f32_16x16x32_fp8_bf8:
-    case Intrinsic::amdgcn_mfma_f32_16x16x32_fp8_fp8:
-    case Intrinsic::amdgcn_mfma_f32_32x32x16_bf8_bf8:
-    case Intrinsic::amdgcn_mfma_f32_32x32x16_bf8_fp8:
-    case Intrinsic::amdgcn_mfma_f32_32x32x16_fp8_bf8:
-    case Intrinsic::amdgcn_mfma_f32_32x32x16_fp8_fp8:
-    case Intrinsic::amdgcn_smfmac_i32_16x16x64_i8:
-    case Intrinsic::amdgcn_smfmac_i32_32x32x32_i8:
-    case Intrinsic::amdgcn_smfmac_f32_16x16x64_bf8_bf8:
-    case Intrinsic::amdgcn_smfmac_f32_16x16x64_bf8_fp8:
-    case Intrinsic::amdgcn_smfmac_f32_16x16x64_fp8_bf8:
-    case Intrinsic::amdgcn_smfmac_f32_16x16x64_fp8_fp8:
-    case Intrinsic::amdgcn_smfmac_f32_32x32x32_bf8_bf8:
-    case Intrinsic::amdgcn_smfmac_f32_32x32x32_bf8_fp8:
-    case Intrinsic::amdgcn_smfmac_f32_32x32x32_fp8_bf8:
-    case Intrinsic::amdgcn_smfmac_f32_32x32x32_fp8_fp8:
-    case Intrinsic::amdgcn_smfmac_i32_16x16x128_i8:
-    case Intrinsic::amdgcn_smfmac_i32_32x32x64_i8:
-    case Intrinsic::amdgcn_smfmac_f32_16x16x128_bf8_bf8:
-    case Intrinsic::amdgcn_smfmac_f32_16x16x128_bf8_fp8:
-    case Intrinsic::amdgcn_smfmac_f32_16x16x128_fp8_bf8:
-    case Intrinsic::amdgcn_smfmac_f32_16x16x128_fp8_fp8:
-    case Intrinsic::amdgcn_smfmac_f32_32x32x64_bf8_bf8:
-    case Intrinsic::amdgcn_smfmac_f32_32x32x64_bf8_fp8:
-    case Intrinsic::amdgcn_smfmac_f32_32x32x64_fp8_bf8:
-    case Intrinsic::amdgcn_smfmac_f32_32x32x64_fp8_fp8:
-    case Intrinsic::amdgcn_wmma_f32_16x16x16_fp8_fp8:
-    case Intrinsic::amdgcn_wmma_f32_16x16x16_fp8_bf8:
-    case Intrinsic::amdgcn_wmma_f32_16x16x16_bf8_fp8:
-    case Intrinsic::amdgcn_wmma_f32_16x16x16_bf8_bf8:
-    case Intrinsic::amdgcn_swmmac_f32_16x16x32_fp8_fp8:
-    case Intrinsic::amdgcn_swmmac_f32_16x16x32_fp8_bf8:
-    case Intrinsic::amdgcn_swmmac_f32_16x16x32_bf8_fp8:
-    case Intrinsic::amdgcn_swmmac_f32_16x16x32_bf8_bf8:
-    case Intrinsic::amdgcn_wmma_i32_16x16x16_iu8:
-    case Intrinsic::amdgcn_wmma_i32_16x16x16_iu4:
-    case Intrinsic::amdgcn_wmma_i32_16x16x32_iu4:
-    case Intrinsic::amdgcn_swmmac_i32_16x16x32_iu8:
-    case Intrinsic::amdgcn_swmmac_i32_16x16x32_iu4:
-    case Intrinsic::amdgcn_swmmac_i32_16x16x64_iu4:
-    case Intrinsic::amdgcn_raw_buffer_store_format:
-    case Intrinsic::amdgcn_raw_buffer_store:
-    case Intrinsic::amdgcn_raw_ptr_buffer_store_format:
-    case Intrinsic::amdgcn_raw_ptr_buffer_store:
-    case Intrinsic::amdgcn_struct_buffer_store_format:
-    case Intrinsic::amdgcn_struct_buffer_store:
-    case Intrinsic::amdgcn_struct_ptr_buffer_store_format:
-    case Intrinsic::amdgcn_struct_ptr_buffer_store:
-    case Intrinsic::amdgcn_raw_tbuffer_store:
-    case Intrinsic::amdgcn_raw_ptr_tbuffer_store:
-    case Intrinsic::amdgcn_struct_ptr_tbuffer_store:
-    case Intrinsic::amdgcn_struct_tbuffer_store:
-      return true;
-    default:
-      return false;
-    }
-  }
-
   bool isOpLegal(Instruction *I) {
-    if (const auto *Intr = dyn_cast<IntrinsicInst>(I)) {
-      Intrinsic::ID ID = Intr->getIntrinsicID();
-      if (isNativeIntrinsic(ID))
-        return true;
-    }
-    // Stores
-    if (isa<StoreInst>(I))
-      return true;
-    return false;
+    return isa<StoreInst>(I) || isa<IntrinsicInst>(I);
   }
 
   bool isCoercionProfitable(Instruction *II) {