61 lines · cpp
1//===- AttrToLLVMConverter.cpp - SPIR-V attributes conversion to LLVM -C++ ===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8 9#include <mlir/Conversion/SPIRVCommon/AttrToLLVMConverter.h>10 11namespace mlir {12namespace {13 14//===----------------------------------------------------------------------===//15// Constants16//===----------------------------------------------------------------------===//17 18constexpr unsigned defaultAddressSpace = 0;19 20//===----------------------------------------------------------------------===//21// Utility functions22//===----------------------------------------------------------------------===//23 24static unsigned25storageClassToOCLAddressSpace(spirv::StorageClass storageClass) {26 // Based on27 // https://registry.khronos.org/SPIR-V/specs/unified1/OpenCL.ExtendedInstructionSet.100.html#_binary_form28 // and clang/lib/Basic/Targets/SPIR.h.29 switch (storageClass) {30 case spirv::StorageClass::Function:31 return 0;32 case spirv::StorageClass::Input:33 case spirv::StorageClass::CrossWorkgroup:34 return 1;35 case spirv::StorageClass::UniformConstant:36 return 2;37 case spirv::StorageClass::Workgroup:38 return 3;39 case spirv::StorageClass::Generic:40 return 4;41 case spirv::StorageClass::DeviceOnlyINTEL:42 return 5;43 case spirv::StorageClass::HostOnlyINTEL:44 return 6;45 default:46 return defaultAddressSpace;47 }48}49} // namespace50 51unsigned storageClassToAddressSpace(spirv::ClientAPI clientAPI,52 spirv::StorageClass storageClass) {53 switch (clientAPI) {54 case spirv::ClientAPI::OpenCL:55 return storageClassToOCLAddressSpace(storageClass);56 default:57 return defaultAddressSpace;58 }59}60} // namespace mlir61