File: PrepareInlineSamplerForBindless.cpp

package info (click to toggle)
intel-graphics-compiler2 2.16.0-2
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 106,644 kB
  • sloc: cpp: 805,640; lisp: 287,672; ansic: 16,414; python: 3,952; yacc: 2,588; lex: 1,666; pascal: 313; sh: 186; makefile: 35
file content (91 lines) | stat: -rw-r--r-- 3,177 bytes parent folder | download
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
/*========================== begin_copyright_notice ============================

Copyright (C) 2025 Intel Corporation

SPDX-License-Identifier: MIT

============================= end_copyright_notice ===========================*/

#include "PrepareInlineSamplerForBindless.hpp"

#include "Compiler/CodeGenPublic.h"
#include "Compiler/IGCPassSupport.h"
#include "Compiler/Optimizer/OCLBIUtils.h"

using namespace IGC;
using namespace llvm;

// Register pass to igc-opt
#define PASS_FLAG "prepare-inline-sampler-for-bindless"
#define PASS_DESCRIPTION "Prepare OCL inline sampler for bindless"
#define PASS_CFG_ONLY false
#define PASS_ANALYSIS false
IGC_INITIALIZE_PASS_BEGIN(PrepareInlineSamplerForBindless, PASS_FLAG, PASS_DESCRIPTION, PASS_CFG_ONLY, PASS_ANALYSIS)
IGC_INITIALIZE_PASS_DEPENDENCY(MetaDataUtilsWrapper)
IGC_INITIALIZE_PASS_DEPENDENCY(CodeGenContextWrapper)
IGC_INITIALIZE_PASS_END(PrepareInlineSamplerForBindless, PASS_FLAG, PASS_DESCRIPTION, PASS_CFG_ONLY, PASS_ANALYSIS)

char PrepareInlineSamplerForBindless::ID = 0;

PrepareInlineSamplerForBindless::PrepareInlineSamplerForBindless() : FunctionPass(ID) {
  initializePrepareInlineSamplerForBindlessPass(*PassRegistry::getPassRegistry());
}

bool PrepareInlineSamplerForBindless::runOnFunction(Function &F) {
  mChanged = false;
  mInlineSamplerIndex = 0;

  if (!mMDUtils) {
    mMDUtils = getAnalysis<MetaDataUtilsWrapper>().getMetaDataUtils();
  }

  if (!isEntryFunc(mMDUtils, &F)) {
    // Only entry functions can be assigned implicit args.
    return false;
  }

  visit(F);

  ImplicitArgs::addImageArgs(F, mArgMap, mMDUtils);
  mArgMap.clear();

  if (mChanged) {
    mMDUtils->save(F.getContext());
  }

  return mChanged;
}

void PrepareInlineSamplerForBindless::visitCallInst(CallInst &CI) {
  Function *Callee = CI.getCalledFunction();
  if (!Callee || Callee->getName() != "__bindless_sampler_initializer") {
    return;
  }

  mChanged = true;

  IGC_ASSERT_MESSAGE(isa<ConstantInt>(CI.getArgOperand(0)),
                     "Sampler initializer calls can only be made with const int values!");

  // Inline sampler doesn't associate with an explicit argument.
  // To avoid adding a new metadata entry, inline sampler value is stored as
  // explicit argument number.
  auto *InlineSamplerInit = cast<ConstantInt>(CI.getArgOperand(0));
  int InlineSamplerInitValue = InlineSamplerInit->getZExtValue();
  auto [_, IsNew] = mArgMap[ImplicitArg::INLINE_SAMPLER].insert(int_cast<int>(InlineSamplerInitValue));
  if (!IsNew) {
    // Sampler initialized by this specific value was already processed.
    // No new sampler will be created. Skip creating inline sampler metadata.
    return;
  }

  // Add metadata for the inline sampler.
  ModuleMetaData *ModMD = getAnalysis<MetaDataUtilsWrapper>().getModuleMetaData();
  FunctionMetaData &FuncMD = ModMD->FuncMD[CI.getFunction()];
  ResourceAllocMD &ResAllocMD = FuncMD.resAllocMD;
  InlineSamplersMD InlineSamplerMD;

  CImagesBI::CreateInlineSamplerAnnotations(CI.getFunction()->getParent(), InlineSamplerMD, InlineSamplerInitValue);
  InlineSamplerMD.index = mInlineSamplerIndex++;
  ResAllocMD.inlineSamplersMD.push_back(InlineSamplerMD);
}