44 lines
1.4 KiB
C++
44 lines
1.4 KiB
C++
//===- PassDetail.cpp - Async Pass class details ----------------*- 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "PassDetail.h"
|
|
#include "mlir/IR/Builders.h"
|
|
#include "mlir/Transforms/RegionUtils.h"
|
|
|
|
using namespace mlir;
|
|
|
|
void mlir::async::cloneConstantsIntoTheRegion(Region ®ion) {
|
|
OpBuilder builder(®ion);
|
|
cloneConstantsIntoTheRegion(region, builder);
|
|
}
|
|
|
|
void mlir::async::cloneConstantsIntoTheRegion(Region ®ion,
|
|
OpBuilder &builder) {
|
|
// Values implicitly captured by the region.
|
|
llvm::SetVector<Value> captures;
|
|
getUsedValuesDefinedAbove(region, region, captures);
|
|
|
|
OpBuilder::InsertionGuard guard(builder);
|
|
builder.setInsertionPointToStart(®ion.front());
|
|
|
|
// Clone ConstantLike operations into the region.
|
|
for (Value capture : captures) {
|
|
Operation *op = capture.getDefiningOp();
|
|
if (!op || !op->hasTrait<OpTrait::ConstantLike>())
|
|
continue;
|
|
|
|
Operation *cloned = builder.clone(*op);
|
|
|
|
for (auto tuple : llvm::zip(op->getResults(), cloned->getResults())) {
|
|
Value orig = std::get<0>(tuple);
|
|
Value replacement = std::get<1>(tuple);
|
|
replaceAllUsesInRegionWith(orig, replacement, region);
|
|
}
|
|
}
|
|
}
|