Valentin Clement (バレンタイン クレメン) bf81bdec66
[mlir][acc] Add isValidValueUse to OpenACCSupport (#171538)
Add a new API `isValidValueUse ` to OpenACCSupport. This is used in
ACCImplicitData to check value that are already legal in the OpenACC
region and do not require implicit clause to be generated. An example
would be a CUDA Fortran device variable that is already on the GPU.
2025-12-10 08:19:04 -08:00

59 lines
2.0 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"
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);
}
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 &region) {
if (impl)
return impl->isValidValueUse(v, region);
return false;
}
} // namespace acc
} // namespace mlir