1/* 2 * Copyright 2016, The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17#ifndef RS2SPIRV_RS_ALLOCATION_UTILS_H 18#define RS2SPIRV_RS_ALLOCATION_UTILS_H 19 20#include "llvm/ADT/Optional.h" 21#include "llvm/ADT/SmallVector.h" 22 23#include <string> 24 25namespace llvm { 26class CallInst; 27class GlobalVariable; 28class Module; 29class Type; 30} // namespace llvm 31 32namespace rs2spirv { 33 34struct RSAllocationInfo { 35 std::string VarName; 36 llvm::Optional<std::string> RSElementType; 37 llvm::GlobalVariable *GlobalVar; 38 // Assigned unique identifier for this allocation; 39 // not the slot #. 40 // Represents the index of this allocation's metadata 41 // in the global allocation metadata Vulkan buffer 42 int ID; 43 bool hasID(void) const { return ID >= 0; } 44 void assignID(int no) { ID = no; } 45}; 46 47enum class RSAllocAccessKind { GEA, SEA, DIMX }; 48 49struct RSAllocationCallInfo { 50 RSAllocationInfo &RSAlloc; 51 llvm::CallInst *FCall; 52 RSAllocAccessKind Kind; 53 std::string RSElementTy; 54}; 55 56bool isRSAllocation(const llvm::GlobalVariable &GV); 57bool getRSAllocationInfo(llvm::Module &M, 58 llvm::SmallVectorImpl<RSAllocationInfo> &Allocs); 59bool getRSAllocAccesses(llvm::SmallVectorImpl<RSAllocationInfo> &Allocs, 60 llvm::SmallVectorImpl<RSAllocationCallInfo> &Calls); 61bool solidifyRSAllocAccess(llvm::Module &M, RSAllocationCallInfo CallInfo); 62 63} // namespace rs2spirv 64 65#endif 66