72c65b698e
TestDialect has many operations and they all live in ::mlir namespace. Sometimes it is not clear whether the ops used in the code for the test passes belong to Standard or to Test dialects. Also, with this change it is easier to understand what test passes registered in mlir-opt are actually passes in mlir/test. Differential Revision: https://reviews.llvm.org/D90794
42 lines
1.4 KiB
C++
42 lines
1.4 KiB
C++
//===- TestGPUParallelLoopMapping.cpp - Test pass for GPU loop mapping ----===//
|
|
//
|
|
// 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 pass testing the utilities for mapping parallel
|
|
// loops to gpu hardware ids.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Dialect/GPU/ParallelLoopMapper.h"
|
|
#include "mlir/Pass/Pass.h"
|
|
|
|
using namespace mlir;
|
|
|
|
namespace {
|
|
/// Simple pass for testing the mapping of parallel loops to hardware ids using
|
|
/// a greedy mapping strategy.
|
|
class TestGpuGreedyParallelLoopMappingPass
|
|
: public PassWrapper<TestGpuGreedyParallelLoopMappingPass,
|
|
OperationPass<FuncOp>> {
|
|
void runOnOperation() override {
|
|
Operation *op = getOperation();
|
|
for (Region ®ion : op->getRegions())
|
|
greedilyMapParallelSCFToGPU(region);
|
|
}
|
|
};
|
|
} // namespace
|
|
|
|
namespace mlir {
|
|
namespace test {
|
|
void registerTestGpuParallelLoopMappingPass() {
|
|
PassRegistration<TestGpuGreedyParallelLoopMappingPass> registration(
|
|
"test-gpu-greedy-parallel-loop-mapping",
|
|
"Greedily maps all parallel loops to gpu hardware ids.");
|
|
}
|
|
} // namespace test
|
|
} // namespace mlir
|