399887c9e4
This commit adds two resource limits, max_compute_workgroup_size and max_compute_workgroup_invocations as resource limits to the target environment. They are not used at the current moment, but they will affect the SPIR-V CodeGen. Adding for now to have a proper target environment modelling. Differential Revision: https://reviews.llvm.org/D73905
90 lines
3.0 KiB
C++
90 lines
3.0 KiB
C++
//===- TargetAndABI.cpp - SPIR-V target and ABI utilities -----------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Dialect/SPIRV/TargetAndABI.h"
|
|
#include "mlir/Dialect/SPIRV/SPIRVTypes.h"
|
|
#include "mlir/IR/Builders.h"
|
|
#include "mlir/IR/FunctionSupport.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);
|
|
}
|
|
|
|
spirv::ResourceLimitsAttr
|
|
spirv::getDefaultResourceLimits(MLIRContext *context) {
|
|
auto i32Type = IntegerType::get(32, context);
|
|
auto v3i32Type = VectorType::get(3, i32Type);
|
|
|
|
// These numbers are from "Table 46. Required Limits" of the Vulkan spec.
|
|
return spirv::ResourceLimitsAttr ::get(
|
|
IntegerAttr::get(i32Type, 128),
|
|
DenseIntElementsAttr::get<int32_t>(v3i32Type, {128, 128, 64}), 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)}),
|
|
spirv::getDefaultResourceLimits(context), context);
|
|
}
|
|
|
|
spirv::TargetEnvAttr spirv::lookupTargetEnvOrDefault(Operation *op) {
|
|
if (auto attr = op->getAttrOfType<spirv::TargetEnvAttr>(
|
|
spirv::getTargetEnvAttrName()))
|
|
return attr;
|
|
return getDefaultTargetEnv(op->getContext());
|
|
}
|
|
|
|
DenseIntElementsAttr spirv::lookupLocalWorkGroupSize(Operation *op) {
|
|
while (op && !op->hasTrait<OpTrait::FunctionLike>())
|
|
op = op->getParentOp();
|
|
if (!op)
|
|
return {};
|
|
|
|
if (auto attr = op->getAttrOfType<spirv::EntryPointABIAttr>(
|
|
spirv::getEntryPointABIAttrName()))
|
|
return attr.local_size();
|
|
|
|
return {};
|
|
}
|