e21adfa32d
This makes ignoring a result explicit by the user, and helps to prevent accidental errors with dropped results. Marking LogicalResult as no discard was always the intention from the beginning, but got lost along the way. Differential Revision: https://reviews.llvm.org/D95841
41 lines
1.4 KiB
C++
41 lines
1.4 KiB
C++
//===- TestAllReduceLowering.cpp - Test gpu.all_reduce lowering -----------===//
|
|
//
|
|
// 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 contains test passes for lowering the gpu.all_reduce op.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Dialect/GPU/Passes.h"
|
|
#include "mlir/Dialect/StandardOps/IR/Ops.h"
|
|
#include "mlir/Pass/Pass.h"
|
|
#include "mlir/Transforms/GreedyPatternRewriteDriver.h"
|
|
|
|
using namespace mlir;
|
|
|
|
namespace {
|
|
struct TestGpuRewritePass
|
|
: public PassWrapper<TestGpuRewritePass, OperationPass<ModuleOp>> {
|
|
void getDependentDialects(DialectRegistry ®istry) const override {
|
|
registry.insert<StandardOpsDialect>();
|
|
}
|
|
void runOnOperation() override {
|
|
OwningRewritePatternList patterns;
|
|
populateGpuRewritePatterns(&getContext(), patterns);
|
|
(void)applyPatternsAndFoldGreedily(getOperation(), std::move(patterns));
|
|
}
|
|
};
|
|
} // namespace
|
|
|
|
namespace mlir {
|
|
void registerTestAllReduceLoweringPass() {
|
|
PassRegistration<TestGpuRewritePass> pass(
|
|
"test-gpu-rewrite",
|
|
"Applies all rewrite patterns within the GPU dialect.");
|
|
}
|
|
} // namespace mlir
|