d45de8003a
Add support in `-convert-gpu-to-llvm-spv` to convert `gpu.func` to `llvm.func` operations. - `spir_kernel`/`spir_func` calling conventions used for kernels/functions. - `workgroup` attributions encoded as additional `llvm.ptr<3>` arguments. - No attribute used to annotate kernels - `reqd_work_group_size` attribute using to encode `gpu.known_block_size`. - `llvm.mlir.workgroup_attrib_size` used to encode workgroup attribution sizes. This will be attached to the pointer argument workgroup attributions lower to. **Note**: A notable missing feature that will be addressed in a follow-up PR is a `-use-bare-ptr-memref-call-conv` option to replace MemRef arguments with bare pointers to the MemRef element types instead of the current MemRef descriptor approach. --------- Signed-off-by: Victor Perez <victor.perez@codeplay.com>
61 lines
1.9 KiB
C++
61 lines
1.9 KiB
C++
//===- AttrToLLVMConverter.cpp - SPIR-V attributes conversion to LLVM -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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include <mlir/Conversion/SPIRVCommon/AttrToLLVMConverter.h>
|
|
|
|
namespace mlir {
|
|
namespace {
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Constants
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
constexpr unsigned defaultAddressSpace = 0;
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Utility functions
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
static unsigned
|
|
storageClassToOCLAddressSpace(spirv::StorageClass storageClass) {
|
|
// Based on
|
|
// https://registry.khronos.org/SPIR-V/specs/unified1/OpenCL.ExtendedInstructionSet.100.html#_binary_form
|
|
// and clang/lib/Basic/Targets/SPIR.h.
|
|
switch (storageClass) {
|
|
case spirv::StorageClass::Function:
|
|
return 0;
|
|
case spirv::StorageClass::Input:
|
|
case spirv::StorageClass::CrossWorkgroup:
|
|
return 1;
|
|
case spirv::StorageClass::UniformConstant:
|
|
return 2;
|
|
case spirv::StorageClass::Workgroup:
|
|
return 3;
|
|
case spirv::StorageClass::Generic:
|
|
return 4;
|
|
case spirv::StorageClass::DeviceOnlyINTEL:
|
|
return 5;
|
|
case spirv::StorageClass::HostOnlyINTEL:
|
|
return 6;
|
|
default:
|
|
return defaultAddressSpace;
|
|
}
|
|
}
|
|
} // namespace
|
|
|
|
unsigned storageClassToAddressSpace(spirv::ClientAPI clientAPI,
|
|
spirv::StorageClass storageClass) {
|
|
switch (clientAPI) {
|
|
case spirv::ClientAPI::OpenCL:
|
|
return storageClassToOCLAddressSpace(storageClass);
|
|
default:
|
|
return defaultAddressSpace;
|
|
}
|
|
}
|
|
} // namespace mlir
|