EnzymeBatchPass.cpp File Reference

#include "Passes/EnzymeBatchPass.h"
#include "Dialect/Ops.h"
#include "Interfaces/GradientUtilsReverse.h"
#include "PassDetails.h"
#include "Passes/Passes.h"
#include "mlir/Dialect/Func/IR/FuncOps.h"
#include "mlir/IR/Builders.h"
#include "mlir/Interfaces/FunctionInterfaces.h"
#include "Passes/Passes.h.inc"

Include dependency graph for EnzymeBatchPass.cpp:

Go to the source code of this file.

Namespaces
namespace	mlir
namespace	mlir::enzyme
namespace	mlir::enzyme::batchutils

Macros
#define	DEBUG_TYPE   "enzyme-batch"
#define	GEN_PASS_DEF_BATCHPASS

Functions
mlir::TensorType	mlir::enzyme::batchutils::applyBatchSizes (mlir::Type Ty, llvm::ArrayRef< int64_t > batchSizes)
LogicalResult	mlir::enzyme::batchutils::handleCallOp (func::CallOp callOp, OpBuilder &builder, IRMapping &mapper, llvm::ArrayRef< int64_t > batchSizes, std::map< BatchCacheKey, FunctionOpInterface > &batchedFunctionCache)
void	mlir::enzyme::batchutils::batchCloneBlock (OpBuilder &builder, Block *blk, IRMapping &mapper, llvm::ArrayRef< int64_t > batchSizes, std::map< BatchCacheKey, FunctionOpInterface > &batchedFunctionCache, bool withoutTerminator)
void	mlir::enzyme::batchutils::batchCloneRegion (OpBuilder &builder, Region *src, Region *dest, IRMapping &mapper, llvm::ArrayRef< int64_t > batchSizes, std::map< BatchCacheKey, FunctionOpInterface > &batchedFunctionCache)
FunctionOpInterface	mlir::enzyme::batchutils::batchCloneFunction (OpBuilder &builder, FunctionOpInterface F, Twine name, llvm::ArrayRef< int64_t > batchSizes, std::map< BatchCacheKey, FunctionOpInterface > &batchedFunctionCache)
template<typename T >
FunctionOpInterface	mlir::enzyme::batchutils::batchOperationWithoutInsertingCallOp (OpBuilder &builder, T CI, FunctionOpInterface fn, std::map< BatchCacheKey, FunctionOpInterface > &batchedFunctionCache)

Macro Definition Documentation

DEBUG_TYPE

#define DEBUG_TYPE   "enzyme-batch"

Definition at line 24 of file EnzymeBatchPass.cpp.

GEN_PASS_DEF_BATCHPASS

#define GEN_PASS_DEF_BATCHPASS

Definition at line 32 of file EnzymeBatchPass.cpp.