mirror of
https://github.com/KhronosGroup/SPIRV-Tools
synced 2024-10-18 11:10:05 +00:00
d997c83b10
This commit adds a spirv-opt pass to replace accesses to descriptor array based on variable indices with constant elements. Before: ``` %descriptor = OpVariable %_ptr_array_Image Uniform ... %ac = OpAccessChain %_ptr_Image %descriptor %variable_index (some image instructions using %ac) ``` After: ``` %descriptor = OpVariable %_ptr_array_Image Uniform ... OpSwitch %variable_index 0 %case0 1 %case1 ... ... %case0 = OpLabel %ac = OpAccessChain %_ptr_Image %descriptor %uint_0 ... %case1 = OpLabel %ac = OpAccessChain %_ptr_Image %descriptor %uint_1 ... (use OpPhi for value with concrete type) ```
55 lines
2.1 KiB
C++
55 lines
2.1 KiB
C++
// Copyright (c) 2021 Google LLC
|
|
//
|
|
// 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 SOURCE_OPT_DESC_SROA_UTIL_H_
|
|
#define SOURCE_OPT_DESC_SROA_UTIL_H_
|
|
|
|
#include "source/opt/ir_context.h"
|
|
|
|
namespace spvtools {
|
|
namespace opt {
|
|
|
|
// Provides functions for the descriptor array SROA.
|
|
namespace descsroautil {
|
|
|
|
// Returns true if |var| is an OpVariable instruction that represents a
|
|
// descriptor array.
|
|
bool IsDescriptorArray(IRContext* context, Instruction* var);
|
|
|
|
// Returns true if |type| is a type that could be used for a structured buffer
|
|
// as opposed to a type that would be used for a structure of resource
|
|
// descriptors.
|
|
bool IsTypeOfStructuredBuffer(IRContext* context, const Instruction* type);
|
|
|
|
// Returns the first index of the OpAccessChain instruction |access_chain| as
|
|
// a constant. Returns nullptr if it is not a constant.
|
|
const analysis::Constant* GetAccessChainIndexAsConst(IRContext* context,
|
|
Instruction* access_chain);
|
|
|
|
// Returns the number of elements of an OpVariable instruction |var| whose type
|
|
// must be a pointer to an array or a struct.
|
|
uint32_t GetNumberOfElementsForArrayOrStruct(IRContext* context,
|
|
Instruction* var);
|
|
|
|
// Returns the first Indexes operand id of the OpAccessChain or
|
|
// OpInBoundsAccessChain instruction |access_chain|. The access chain must have
|
|
// at least 1 index.
|
|
uint32_t GetFirstIndexOfAccessChain(Instruction* access_chain);
|
|
|
|
} // namespace descsroautil
|
|
} // namespace opt
|
|
} // namespace spvtools
|
|
|
|
#endif // SOURCE_OPT_DESC_SROA_UTIL_H_
|