47c6ab2b97
This commit defines a new SPIR-V dialect attribute for specifying a SPIR-V target environment. It is a dictionary attribute containing the SPIR-V version, supported extension list, and allowed capability list. A SPIRVConversionTarget subclass is created to take in the target environment and sets proper dynmaically legal ops by querying the op availability interface of SPIR-V ops to make sure they are available in the specified target environment. All existing conversions targeting SPIR-V is changed to use this SPIRVConversionTarget. It probes whether the input IR has a `spv.target_env` attribute, otherwise, it uses the default target environment: SPIR-V 1.0 with Shader capability and no extra extensions. Differential Revision: https://reviews.llvm.org/D72256
65 lines
2.2 KiB
C++
65 lines
2.2 KiB
C++
//===- TargetAndABI.cpp - SPIR-V target and ABI utilities -----------------===//
|
|
//
|
|
// Part of the MLIR 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Dialect/SPIRV/TargetAndABI.h"
|
|
#include "mlir/Dialect/SPIRV/SPIRVTypes.h"
|
|
#include "mlir/IR/Builders.h"
|
|
#include "mlir/IR/Operation.h"
|
|
|
|
using namespace mlir;
|
|
|
|
namespace mlir {
|
|
#include "mlir/Dialect/SPIRV/TargetAndABI.cpp.inc"
|
|
}
|
|
|
|
StringRef spirv::getInterfaceVarABIAttrName() {
|
|
return "spv.interface_var_abi";
|
|
}
|
|
|
|
spirv::InterfaceVarABIAttr
|
|
spirv::getInterfaceVarABIAttr(unsigned descriptorSet, unsigned binding,
|
|
spirv::StorageClass storageClass,
|
|
MLIRContext *context) {
|
|
Type i32Type = IntegerType::get(32, context);
|
|
return spirv::InterfaceVarABIAttr::get(
|
|
IntegerAttr::get(i32Type, descriptorSet),
|
|
IntegerAttr::get(i32Type, binding),
|
|
IntegerAttr::get(i32Type, static_cast<int64_t>(storageClass)), context);
|
|
}
|
|
|
|
StringRef spirv::getEntryPointABIAttrName() { return "spv.entry_point_abi"; }
|
|
|
|
spirv::EntryPointABIAttr
|
|
spirv::getEntryPointABIAttr(ArrayRef<int32_t> localSize, MLIRContext *context) {
|
|
assert(localSize.size() == 3);
|
|
return spirv::EntryPointABIAttr::get(
|
|
DenseElementsAttr::get<int32_t>(
|
|
VectorType::get(3, IntegerType::get(32, context)), localSize)
|
|
.cast<DenseIntElementsAttr>(),
|
|
context);
|
|
}
|
|
|
|
StringRef spirv::getTargetEnvAttrName() { return "spv.target_env"; }
|
|
|
|
spirv::TargetEnvAttr spirv::getDefaultTargetEnv(MLIRContext *context) {
|
|
Builder builder(context);
|
|
return spirv::TargetEnvAttr::get(
|
|
builder.getI32IntegerAttr(static_cast<uint32_t>(spirv::Version::V_1_0)),
|
|
builder.getI32ArrayAttr({}),
|
|
builder.getI32ArrayAttr(
|
|
{static_cast<uint32_t>(spirv::Capability::Shader)}),
|
|
context);
|
|
}
|
|
|
|
spirv::TargetEnvAttr spirv::lookupTargetEnvOrDefault(Operation *op) {
|
|
if (auto attr = op->getAttrOfType<spirv::TargetEnvAttr>(
|
|
spirv::getTargetEnvAttrName()))
|
|
return attr;
|
|
return getDefaultTargetEnv(op->getContext());
|
|
}
|