LoopsToGPU.h

Go to the documentation of this file.

//===- LoopsToGPU.h - Convert loop nests to GPU kernels ---------*- C++ -*-===//
//
// Part of the MLIR 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
//
//===----------------------------------------------------------------------===//
#ifndef MLIR_CONVERSION_LOOPSTOGPU_LOOPSTOGPU_H_
#define MLIR_CONVERSION_LOOPSTOGPU_LOOPSTOGPU_H_

#include "mlir/Support/LLVM.h"

namespace mlir {
class AffineForOp;
struct LogicalResult;
class Value;

namespace loop {
class ForOp;
} // end namespace loop

LogicalResult convertAffineLoopNestToGPULaunch(AffineForOp forOp,
                                               unsigned numBlockDims,
                                               unsigned numThreadDims);

LogicalResult convertLoopNestToGPULaunch(loop::ForOp forOp,
                                         unsigned numBlockDims,
                                         unsigned numThreadDims);

LogicalResult convertLoopToGPULaunch(loop::ForOp forOp,
                                     ArrayRef<Value> numWorkGroups,
                                     ArrayRef<Value> workGroupSizes);

} // namespace mlir

#endif // MLIR_CONVERSION_LOOPSTOGPU_LOOPSTOGPU_H_