DataLayout.h

//===-- Optimizer/Support/DataLayout.h --------------------------*- C++ -*-===//
//
// Part of the LLVM 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
//
//===----------------------------------------------------------------------===//
//
// Coding style: https://mlir.llvm.org/getting_started/DeveloperGuide/
//
//===----------------------------------------------------------------------===//
 
#ifndef FORTRAN_OPTIMIZER_SUPPORT_DATALAYOUT_H
#define FORTRAN_OPTIMIZER_SUPPORT_DATALAYOUT_H
 
#include "mlir/Interfaces/DataLayoutInterfaces.h"
#include <optional>
 
namespace mlir {
class ModuleOp;
namespace gpu {
class GPUModuleOp;
} // namespace gpu
} // namespace mlir
 
namespace llvm {
class DataLayout;
} // namespace llvm
 
namespace fir::support {
void setMLIRDataLayout(mlir::ModuleOp mlirModule, const llvm::DataLayout &dl);
void setMLIRDataLayout(mlir::gpu::GPUModuleOp mlirModule,
                       const llvm::DataLayout &dl);
 
void setMLIRDataLayoutFromAttributes(mlir::ModuleOp mlirModule,
                                     bool allowDefaultLayout);
void setMLIRDataLayoutFromAttributes(mlir::gpu::GPUModuleOp mlirModule,
                                     bool allowDefaultLayout);
 
std::optional<mlir::DataLayout>
getOrSetMLIRDataLayout(mlir::ModuleOp mlirModule,
                       bool allowDefaultLayout = false);
std::optional<mlir::DataLayout>
getOrSetMLIRDataLayout(mlir::gpu::GPUModuleOp mlirModule,
                       bool allowDefaultLayout = false);
 
} // namespace fir::support
 
#endif // FORTRAN_OPTIMIZER_SUPPORT_DATALAYOUT_H