|
MLIR
22.0.0git
|
Options that allow distribution of loops generated in Linalg transforms to processors while generating the loops. More...
#include "mlir/Dialect/Linalg/Utils/Utils.h"
Public Attributes | |
| ProcInfoCallBackFn | procInfo |
Callback function that returns the Values for processor ID (procId), and number of processors (nprocs) used to execute the parallel loops. More... | |
Options that allow distribution of loops generated in Linalg transforms to processors while generating the loops.
| ProcInfoCallBackFn mlir::linalg::LinalgLoopDistributionOptions::procInfo |
Callback function that returns the Values for processor ID (procId), and number of processors (nprocs) used to execute the parallel loops.
The number of {procId, nprocs} pairs returned must be equal to the number of parallelLoopRanges passed into the callback. The parallelLoopRanges are ranges of the outer parallel loops of the operation that do have non-zero tile sizes specified.