7154e75185
Add utility functions to support OpenACC code generation: - getOrCreateGPUModule for GPU module management with configurable naming - getDataLayout for querying data layout from operations with explicit spec or module fallback
77 lines
2.6 KiB
C++
77 lines
2.6 KiB
C++
//===- OpenACCSupport.cpp - OpenACCSupport Implementation -----------------===//
|
|
//
|
|
// 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 file implements the OpenACCSupport analysis interface.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Dialect/OpenACC/Analysis/OpenACCSupport.h"
|
|
#include "mlir/Dialect/OpenACC/OpenACCUtils.h"
|
|
#include "mlir/Dialect/OpenACC/OpenACCUtilsGPU.h"
|
|
|
|
namespace mlir {
|
|
namespace acc {
|
|
|
|
std::string OpenACCSupport::getVariableName(Value v) {
|
|
if (impl)
|
|
return impl->getVariableName(v);
|
|
return acc::getVariableName(v);
|
|
}
|
|
|
|
std::string OpenACCSupport::getRecipeName(RecipeKind kind, Type type,
|
|
Value var) {
|
|
if (impl)
|
|
return impl->getRecipeName(kind, type, var);
|
|
// The default implementation assumes that only type matters
|
|
// and the actual instance of variable is not relevant.
|
|
auto recipeName = acc::getRecipeName(kind, type);
|
|
if (recipeName.empty())
|
|
emitNYI(var ? var.getLoc() : UnknownLoc::get(type.getContext()),
|
|
"variable privatization (incomplete recipe name handling)");
|
|
return recipeName;
|
|
}
|
|
|
|
InFlightDiagnostic OpenACCSupport::emitNYI(Location loc, const Twine &message) {
|
|
if (impl)
|
|
return impl->emitNYI(loc, message);
|
|
return mlir::emitError(loc, "not yet implemented: " + message);
|
|
}
|
|
|
|
remark::detail::InFlightRemark
|
|
OpenACCSupport::emitRemark(Operation *op,
|
|
std::function<std::string()> messageFn,
|
|
llvm::StringRef category) {
|
|
if (impl)
|
|
return impl->emitRemark(op, std::move(messageFn), category);
|
|
return acc::emitRemark(op, messageFn(), category);
|
|
}
|
|
|
|
bool OpenACCSupport::isValidSymbolUse(Operation *user, SymbolRefAttr symbol,
|
|
Operation **definingOpPtr) {
|
|
if (impl)
|
|
return impl->isValidSymbolUse(user, symbol, definingOpPtr);
|
|
return acc::isValidSymbolUse(user, symbol, definingOpPtr);
|
|
}
|
|
|
|
bool OpenACCSupport::isValidValueUse(Value v, Region ®ion) {
|
|
if (impl)
|
|
return impl->isValidValueUse(v, region);
|
|
return acc::isValidValueUse(v, region);
|
|
}
|
|
|
|
std::optional<gpu::GPUModuleOp>
|
|
OpenACCSupport::getOrCreateGPUModule(ModuleOp mod, bool create,
|
|
llvm::StringRef name) {
|
|
if (impl)
|
|
return impl->getOrCreateGPUModule(mod, create, name);
|
|
return acc::getOrCreateGPUModule(mod, create, name);
|
|
}
|
|
|
|
} // namespace acc
|
|
} // namespace mlir
|