llvm-project/llvm/lib/Target/SPIRV/SPIRVInlineAsmLowering.cpp
Vyacheslav Levytskyy 214e6b40f8
[SPIR-V] Inline assembly support (#93164)
This PR introduces support for inline assembly calls for SPIR-V Backend
in general, and support for SPV_INTEL_inline_assembly [1] extension in
particular. The former part of the PR is agnostic towards
vendor-specific requirements and resolves the task of supporting
successful transformation of inline assembly as long as it's possible
without specific SPIR-V instruction codes.

As a part of the PR there appears an opportunity to bring coherent
inline assembly information up to latest passes of the transformation
process (emitting final SPIR-V instructions), so that PR makes it easy
to add any another required flavor of inline assembly, other then
supported by the vendor specific SPV_INTEL_inline_assembly extension,
if/when needed.

At the moment, however, SPV_INTEL_inline_assembly is the only
implemented way to bring LLVM IR inline assembly calls up to valid
SPIR-V instructions and also the default one. This means that inline
assembly calls will generate an error message of such extension is not
used to prevent LLVM-generated error messages at the final stages of
translation. When the SPV_INTEL_inline_assembly extension is mentioned
among supported, translation of inline assembly is intercepted by this
extension implementation on a pre-legalizer step, and this is a place
where support for a new inline assembly extension may be added if
needed.

This PR also extends support for register classes, improves type
inference during pre-legalizer pass, and fixes a minor bug with
asm-printing of string literals.

[1]
https://github.com/intel/llvm/blob/sycl/sycl/doc/design/spirv-extensions/SPV_INTEL_inline_assembly.asciidoc
2024-05-24 15:15:03 +02:00

47 lines
1.7 KiB
C++

//===--- SPIRVInlineAsmLowering.cpp - Inline Asm lowering -------*- 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
//
//===----------------------------------------------------------------------===//
//
// This file implements the lowering of LLVM inline asm calls to machine code
// calls for GlobalISel.
//
//===----------------------------------------------------------------------===//
#include "SPIRVInlineAsmLowering.h"
#include "SPIRVSubtarget.h"
#include "llvm/IR/IntrinsicInst.h"
#include "llvm/IR/IntrinsicsSPIRV.h"
using namespace llvm;
SPIRVInlineAsmLowering::SPIRVInlineAsmLowering(const SPIRVTargetLowering &TLI)
: InlineAsmLowering(&TLI) {}
bool SPIRVInlineAsmLowering::lowerAsmOperandForConstraint(
Value *Val, StringRef Constraint, std::vector<MachineOperand> &Ops,
MachineIRBuilder &MIRBuilder) const {
Value *ValOp = nullptr;
if (isa<ConstantInt>(Val)) {
ValOp = Val;
} else if (ConstantFP *CFP = dyn_cast<ConstantFP>(Val)) {
Ops.push_back(MachineOperand::CreateFPImm(CFP));
return true;
} else if (auto *II = dyn_cast<IntrinsicInst>(Val)) {
if (II->getIntrinsicID() == Intrinsic::spv_track_constant) {
if (isa<ConstantInt>(II->getOperand(0))) {
ValOp = II->getOperand(0);
} else if (ConstantFP *CFP = dyn_cast<ConstantFP>(II->getOperand(0))) {
Ops.push_back(MachineOperand::CreateFPImm(CFP));
return true;
}
}
}
return ValOp ? InlineAsmLowering::lowerAsmOperandForConstraint(
ValOp, Constraint, Ops, MIRBuilder)
: false;
}