This patch extends the GPU kernel outlining pass so that it can take in an optional data layout specification that will be attached to the GPU module operation generated. If the data layout specification is not provided the default data layout is used instead. Reviewed By: herhut, mehdi_amini Differential Revision: https://reviews.llvm.org/D115722
24 lines
720 B
C++
24 lines
720 B
C++
//===- PassDetail.h - GPU Pass class details --------------------*- C++ -*-===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef DIALECT_GPU_TRANSFORMS_PASSDETAIL_H_
|
|
#define DIALECT_GPU_TRANSFORMS_PASSDETAIL_H_
|
|
|
|
#include "mlir/Dialect/Async/IR/Async.h"
|
|
#include "mlir/Dialect/DLTI/DLTI.h"
|
|
#include "mlir/Pass/Pass.h"
|
|
|
|
namespace mlir {
|
|
|
|
#define GEN_PASS_CLASSES
|
|
#include "mlir/Dialect/GPU/Passes.h.inc"
|
|
|
|
} // namespace mlir
|
|
|
|
#endif // DIALECT_GPU_TRANSFORMS_PASSDETAIL_H_
|