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