commit 698c800142d84d32471f36c7e0d33e68db07e06f
author Ilia Diachkov <ilia.diachkov@gmail.com> Wed Aug 31 14:53:02 2022 +0300
committer Ilia Diachkov <ilia.diachkov@gmail.com> Thu Sep 01 16:44:54 2022 +0300
tree 84abc8ac550eb7ee6b238f7f8e5b274c98e824e5
parent c3d1504d6316c11164e5edb9415e6fb43bb705f3

[SPIRV] support builtin types and ExtInsts selection

The patch adds the support of OpenCL and SPIR-V built-in types. It also
implements ExtInst selection and adds spv_unreachable and spv_alloca
intrinsics which improve the generation of the corresponding SPIR-V code.
Five LIT tests are included to demonstrate the improvement.

Differential Revision: https://reviews.llvm.org/D132648

Co-authored-by: Aleksandr Bezzubikov <zuban32s@gmail.com>
Co-authored-by: Michal Paszkowski <michal.paszkowski@outlook.com>
Co-authored-by: Andrey Tretyakov <andrey1.tretyakov@intel.com>
Co-authored-by: Konrad Trifunovic <konrad.trifunovic@intel.com>

llvm/lib/Target/SPIRV/SPIRVLegalizerInfo.cpp

diff --git a/llvm/lib/Target/SPIRV/SPIRVLegalizerInfo.cpp b/llvm/lib/Target/SPIRV/SPIRVLegalizerInfo.cpp
index 87f9e95..3d12077 100644
--- a/llvm/lib/Target/SPIRV/SPIRVLegalizerInfo.cpp
+++ b/llvm/lib/Target/SPIRV/SPIRVLegalizerInfo.cpp
@@ -255,6 +255,18 @@
   getActionDefinitionsBuilder(G_FPOWI).legalForCartesianProduct(
       allFloatScalarsAndVectors, allIntScalarsAndVectors);
 
+  if (ST.canUseExtInstSet(SPIRV::InstructionSet::OpenCL_std)) {
+    getActionDefinitionsBuilder(G_FLOG10).legalFor(allFloatScalarsAndVectors);
+
+    getActionDefinitionsBuilder(
+        {G_CTTZ, G_CTTZ_ZERO_UNDEF, G_CTLZ, G_CTLZ_ZERO_UNDEF})
+        .legalForCartesianProduct(allIntScalarsAndVectors,
+                                  allIntScalarsAndVectors);
+
+    // Struct return types become a single scalar, so cannot easily legalize.
+    getActionDefinitionsBuilder({G_SMULH, G_UMULH}).alwaysLegal();
+  }
+
   getLegacyLegalizerInfo().computeTables();
   verify(*ST.getInstrInfo());
 }