dbd00a5968
This PR improves type inference of operand presented by opaque pointers and aggregate types: * tries to restore original function return type for aggregate types so that it's possible to deduce a correct type during emit-intrinsics step (see llvm/test/CodeGen/SPIRV/SpecConstants/restore-spec-type.ll for the reproducer of the previously existed issue when spirv-val found a mismatch between object and ptr types in OpStore due to the incorrect aggregate types tracing), * explores untyped pointer operands of store to deduce correct pointee types, * creates an extension type to track pointee types from emit-intrinsics step and further instead of direct and naive usage of TypePointerType that led previously to crashes due to ban of creation of Value of TypePointerType type, * tracks instructions with uncomplete type information and tries to improve their type info after pass calculated types for all machine functions (it doesn't traverse a code but rather checks only those instructions which were tracked as uncompleted), * address more cases of removing unnecessary bitcasts (see, for example, changes in test/CodeGen/SPIRV/transcoding/OpGenericCastToPtr.ll where `CHECK-SPIRV-NEXT` in LIT checks show absence of unneeded bitcasts and unmangled/mangled versions have proper typing now with equivalent type info), * address more cases of well known types or relations between types within instructions (see, for example, atomic*.ll test cases and Event-related test cases for improved SPIR-V code generated by the Backend), * fix the issue of removing unneeded ptrcast instructions in pre-legalizer pass that led to creation of new assign-type instructions with the same argument as source in ptrcast and caused errors in type inference (the reproducer `complex.ll` test case is added to the PR).
85 lines
3.8 KiB
C++
85 lines
3.8 KiB
C++
//===-- SPIRVBuiltins.h - SPIR-V Built-in Functions -------------*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Lowering builtin function calls and types using their demangled names.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_LIB_TARGET_SPIRV_SPIRVBUILTINS_H
|
|
#define LLVM_LIB_TARGET_SPIRV_SPIRVBUILTINS_H
|
|
|
|
#include "SPIRVGlobalRegistry.h"
|
|
#include "llvm/CodeGen/GlobalISel/CallLowering.h"
|
|
#include "llvm/CodeGen/GlobalISel/MachineIRBuilder.h"
|
|
|
|
namespace llvm {
|
|
namespace SPIRV {
|
|
/// Parses the name part of the demangled builtin call.
|
|
std::string lookupBuiltinNameHelper(StringRef DemangledCall);
|
|
/// Lowers a builtin function call using the provided \p DemangledCall skeleton
|
|
/// and external instruction \p Set.
|
|
///
|
|
/// \return the lowering success status if the called function is a recognized
|
|
/// builtin, std::nullopt otherwise.
|
|
///
|
|
/// \p DemangledCall is the skeleton of the lowered builtin function call.
|
|
/// \p Set is the external instruction set containing the given builtin.
|
|
/// \p OrigRet is the single original virtual return register if defined,
|
|
/// Register(0) otherwise.
|
|
/// \p OrigRetTy is the type of the \p OrigRet.
|
|
/// \p Args are the arguments of the lowered builtin call.
|
|
std::optional<bool> lowerBuiltin(const StringRef DemangledCall,
|
|
InstructionSet::InstructionSet Set,
|
|
MachineIRBuilder &MIRBuilder,
|
|
const Register OrigRet, const Type *OrigRetTy,
|
|
const SmallVectorImpl<Register> &Args,
|
|
SPIRVGlobalRegistry *GR);
|
|
|
|
/// Helper function for finding a builtin function attributes
|
|
/// by a demangled function name. Defined in SPIRVBuiltins.cpp.
|
|
std::tuple<int, unsigned, unsigned>
|
|
mapBuiltinToOpcode(const StringRef DemangledCall,
|
|
SPIRV::InstructionSet::InstructionSet Set);
|
|
|
|
/// Parses the provided \p ArgIdx argument base type in the \p DemangledCall
|
|
/// skeleton. A base type is either a basic type (e.g. i32 for int), pointer
|
|
/// element type (e.g. i8 for char*), or builtin type (TargetExtType).
|
|
///
|
|
/// \return LLVM Type or nullptr if unrecognized
|
|
///
|
|
/// \p DemangledCall is the skeleton of the lowered builtin function call.
|
|
/// \p ArgIdx is the index of the argument to parse.
|
|
Type *parseBuiltinCallArgumentBaseType(const StringRef DemangledCall,
|
|
unsigned ArgIdx, LLVMContext &Ctx);
|
|
|
|
/// Translates a string representing a SPIR-V or OpenCL builtin type to a
|
|
/// TargetExtType that can be further lowered with lowerBuiltinType().
|
|
///
|
|
/// \return A TargetExtType representing the builtin SPIR-V type.
|
|
///
|
|
/// \p TypeName is the full string representation of the SPIR-V or OpenCL
|
|
/// builtin type.
|
|
TargetExtType *parseBuiltinTypeNameToTargetExtType(std::string TypeName,
|
|
LLVMContext &Context);
|
|
|
|
/// Handles the translation of the provided special opaque/builtin type \p Type
|
|
/// to SPIR-V type. Generates the corresponding machine instructions for the
|
|
/// target type or gets the already existing OpType<...> register from the
|
|
/// global registry \p GR.
|
|
///
|
|
/// \return A machine instruction representing the OpType<...> SPIR-V type.
|
|
///
|
|
/// \p Type is the special opaque/builtin type to be lowered.
|
|
SPIRVType *lowerBuiltinType(const Type *Type,
|
|
AccessQualifier::AccessQualifier AccessQual,
|
|
MachineIRBuilder &MIRBuilder,
|
|
SPIRVGlobalRegistry *GR);
|
|
} // namespace SPIRV
|
|
} // namespace llvm
|
|
#endif // LLVM_LIB_TARGET_SPIRV_SPIRVBUILTINS_H
|