blob: 921408323ab0e4ff1cd574b7e95dad941980da20 [file] [log] [blame]
//===- Passes.h - Pass Entrypoints ------------------------------*- C++ -*-===//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// This header file defines utility functions exposed by the GPU dialect
#include "mlir/Support/LLVM.h"
namespace mlir {
struct LogicalResult;
class Value;
namespace gpu {
class GPUFuncOp;
class LaunchOp;
} // namespace gpu
/// Get a gpu.func created from outlining the region of a gpu.launch op with the
/// given `kernelFnName`. The region of the `launchOp` can use values from
/// above. These need to be captured and passed as arguments to the generated
/// gpu.func. The generated function has arguments
/// - corresponding to the values passed in as `operands`, in that order.
/// - any additional values that might be used within the region of the
/// `launchOp` and defined above it. These captured values are appended to the
/// `operands` list.
gpu::GPUFuncOp outlineKernelFunc(gpu::LaunchOp launchOp, StringRef kernelFnName,
SmallVectorImpl<Value> &operands);
/// Sink operations into the `launchOp` to reduce the number of values that are
/// used within the region of the operation, but defined outside of the
/// region.
LogicalResult sinkOperationsIntoLaunchOp(gpu::LaunchOp launchOp);
} // namespace mlir