MLIR  18.0.0git
Public Member Functions | Public Attributes | List of all members
mlir::scf::SCFTilingOptions Struct Reference

Options to use to control tiling. More...

#include "mlir/Dialect/SCF/Transforms/TileUsingInterface.h"

Public Member Functions

SCFTilingOptionssetTileSizeComputationFunction (SCFTileSizeComputationFunction fun)
 
SCFTilingOptionssetTileSizes (ArrayRef< OpFoldResult > ts)
 Convenience function to set the tileSizeComputationFunction to a function that computes tile sizes at the point they are needed. More...
 
SCFTilingOptionssetInterchange (ArrayRef< int64_t > interchange)
 
SCFTilingOptionssetMapping (ArrayRef< DeviceMappingAttrInterface > mapping)
 

Public Attributes

SCFTileSizeComputationFunction tileSizeComputationFunction = nullptr
 Computation function that returns the tile sizes for each operation. More...
 
SmallVector< int64_t > interchangeVector = {}
 The interchange vector to reorder the tiled loops. More...
 
SmallVector< AttributemappingVector = {}
 Specify mapping of loops to devices. More...
 

Detailed Description

Options to use to control tiling.

Definition at line 32 of file TileUsingInterface.h.

Member Function Documentation

◆ setInterchange()

SCFTilingOptions& mlir::scf::SCFTilingOptions::setInterchange ( ArrayRef< int64_t >  interchange)
inline

Definition at line 50 of file TileUsingInterface.h.

References interchangeVector.

◆ setMapping()

SCFTilingOptions& mlir::scf::SCFTilingOptions::setMapping ( ArrayRef< DeviceMappingAttrInterface >  mapping)
inline

Definition at line 60 of file TileUsingInterface.h.

References mappingVector.

◆ setTileSizeComputationFunction()

SCFTilingOptions& mlir::scf::SCFTilingOptions::setTileSizeComputationFunction ( SCFTileSizeComputationFunction  fun)
inline

Definition at line 39 of file TileUsingInterface.h.

References tileSizeComputationFunction.

◆ setTileSizes()

scf::SCFTilingOptions & scf::SCFTilingOptions::setTileSizes ( ArrayRef< OpFoldResult ts)

Convenience function to set the tileSizeComputationFunction to a function that computes tile sizes at the point they are needed.

Allows proper interaction with folding.

Definition at line 34 of file TileUsingInterface.cpp.

References tileSizeComputationFunction.

Member Data Documentation

◆ interchangeVector

SmallVector<int64_t> mlir::scf::SCFTilingOptions::interchangeVector = {}

The interchange vector to reorder the tiled loops.

Definition at line 49 of file TileUsingInterface.h.

Referenced by setInterchange().

◆ mappingVector

SmallVector<Attribute> mlir::scf::SCFTilingOptions::mappingVector = {}

Specify mapping of loops to devices.

This is only respected when the loop constructs support such a mapping (like scf.forall). Will be ignored when using loop constructs that dont support such a mapping (like scf.for)

Definition at line 59 of file TileUsingInterface.h.

Referenced by setMapping().

◆ tileSizeComputationFunction

SCFTileSizeComputationFunction mlir::scf::SCFTilingOptions::tileSizeComputationFunction = nullptr

Computation function that returns the tile sizes for each operation.

Delayed construction of constant tile sizes should occur to interoperate with folding.

Definition at line 36 of file TileUsingInterface.h.

Referenced by setTileSizeComputationFunction(), and setTileSizes().


The documentation for this struct was generated from the following files: