c64770506b
In the previous state, we were relying on forcing the linker to include all libraries in the final binary and the global initializer to self-register every piece of the system. This change help moving away from this model, and allow users to compose pieces more freely. The current change is only "fixing" the dialect registration and avoiding relying on "whole link" for the passes. The translation is still relying on the global registry, and some refactoring is needed to make this all more convenient. Differential Revision: https://reviews.llvm.org/D74461
45 lines
1.7 KiB
C++
45 lines
1.7 KiB
C++
//===- TestGPUMemoryPromotionPass.cpp - Test pass for GPU promotion -------===//
|
|
//
|
|
// 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 moving data across
|
|
// different levels of the GPU memory hierarchy.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Dialect/GPU/GPUDialect.h"
|
|
#include "mlir/Dialect/GPU/MemoryPromotion.h"
|
|
#include "mlir/IR/Attributes.h"
|
|
#include "mlir/Pass/Pass.h"
|
|
|
|
using namespace mlir;
|
|
|
|
namespace {
|
|
/// Simple pass for testing the promotion to workgroup memory in GPU functions.
|
|
/// Promotes all arguments with "gpu.test_promote_workgroup" attribute. This
|
|
/// does not check whether the promotion is legal (e.g., amount of memory used)
|
|
/// or beneficial (e.g., makes previously uncoalesced loads coalesced).
|
|
class TestGpuMemoryPromotionPass
|
|
: public OperationPass<TestGpuMemoryPromotionPass, gpu::GPUFuncOp> {
|
|
void runOnOperation() override {
|
|
gpu::GPUFuncOp op = getOperation();
|
|
for (unsigned i = 0, e = op.getNumArguments(); i < e; ++i) {
|
|
if (op.getArgAttrOfType<UnitAttr>(i, "gpu.test_promote_workgroup"))
|
|
promoteToWorkgroupMemory(op, i);
|
|
}
|
|
}
|
|
};
|
|
} // end namespace
|
|
|
|
namespace mlir {
|
|
void registerTestGpuMemoryPromotionPass() {
|
|
PassRegistration<TestGpuMemoryPromotionPass>(
|
|
"test-gpu-memory-promotion",
|
|
"Promotes the annotated arguments of gpu.func to workgroup memory.");
|
|
}
|
|
} // namespace mlir
|