TheRock/compiler/amd-llvm/clang/lib/CIR/CodeGen/CIRGenOpenACCHelpers.h

//===----------------------------------------------------------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
//
// This contains helpers for OpenACC emission that don't need to be in
// CIRGenModule, but can't live in a single .cpp file.
//
//===----------------------------------------------------------------------===//
#include "mlir/Dialect/OpenACC/OpenACC.h"
#include "clang/AST/DeclOpenACC.h"

namespace clang::CIRGen {
inline mlir::acc::DataClauseModifier
convertOpenACCModifiers(OpenACCModifierKind modifiers) {
  using namespace mlir::acc;
  static_assert(static_cast<int>(OpenACCModifierKind::Zero) ==
                    static_cast<int>(DataClauseModifier::zero) &&
                static_cast<int>(OpenACCModifierKind::Readonly) ==
                    static_cast<int>(DataClauseModifier::readonly) &&
                static_cast<int>(OpenACCModifierKind::AlwaysIn) ==
                    static_cast<int>(DataClauseModifier::alwaysin) &&
                static_cast<int>(OpenACCModifierKind::AlwaysOut) ==
                    static_cast<int>(DataClauseModifier::alwaysout) &&
                static_cast<int>(OpenACCModifierKind::Capture) ==
                    static_cast<int>(DataClauseModifier::capture));

  DataClauseModifier mlirModifiers{};

  // The MLIR representation of this represents `always` as `alwaysin` +
  // `alwaysout`.  So do a small fixup here.
  if (isOpenACCModifierBitSet(modifiers, OpenACCModifierKind::Always)) {
    mlirModifiers = mlirModifiers | DataClauseModifier::always;
    modifiers &= ~OpenACCModifierKind::Always;
  }

  mlirModifiers = mlirModifiers | static_cast<DataClauseModifier>(modifiers);
  return mlirModifiers;
}

inline mlir::acc::DeviceType decodeDeviceType(const IdentifierInfo *ii) {
  // '*' case leaves no identifier-info, just a nullptr.
  if (!ii)
    return mlir::acc::DeviceType::Star;
  return llvm::StringSwitch<mlir::acc::DeviceType>(ii->getName())
      .CaseLower("default", mlir::acc::DeviceType::Default)
      .CaseLower("host", mlir::acc::DeviceType::Host)
      .CaseLower("multicore", mlir::acc::DeviceType::Multicore)
      .CasesLower({"nvidia", "acc_device_nvidia"},
                  mlir::acc::DeviceType::Nvidia)
      .CaseLower("radeon", mlir::acc::DeviceType::Radeon);
}
} // namespace clang::CIRGen