LayoutUtils.h

Go to the documentation of this file.

//===-- LayoutUtils.h - Decorate composite type with layout information ---===//
//
// Part of the MLIR Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
// This file defines utilities used to get alignment and layout information for
// types in SPIR-V dialect.
//
//===----------------------------------------------------------------------===//
#ifndef MLIR_DIALECT_SPIRV_LAYOUTUTILS_H_
#define MLIR_DIALECT_SPIRV_LAYOUTUTILS_H_

#include <cstdint>

namespace mlir {
class Type;
class VectorType;
namespace spirv {
class StructType;
class ArrayType;
} // namespace spirv

class VulkanLayoutUtils {
public:
  using Size = uint64_t;

  static spirv::StructType decorateType(spirv::StructType structType,
                                        Size &size, Size &alignment);
  static bool isLegalType(Type type);

private:
  static Type decorateType(Type type, Size &size, Size &alignment);
  static Type decorateType(VectorType vectorType, Size &size, Size &alignment);
  static Type decorateType(spirv::ArrayType arrayType, Size &size,
                           Size &alignment);
  static Size getScalarTypeAlignment(Type scalarType);
};

} // namespace mlir

#endif // MLIR_DIALECT_SPIRV_LAYOUTUTILS_H_