Skip to content

Commit

Permalink
Adding comments to explain blockSizeOverride
Browse files Browse the repository at this point in the history
  • Loading branch information
jerryyin committed Jun 24, 2020
1 parent 8448cec commit ffc7951
Showing 1 changed file with 11 additions and 0 deletions.
11 changes: 11 additions & 0 deletions mlir/lib/Dialect/MIOpen/Transforms/AffixTuningParameters.cpp
Expand Up @@ -25,6 +25,17 @@ struct AffixTuningParameters : public MIOpenOpsAffixTuningParametersPassBase<Aff

private:
LaunchDimensionCallback launchDimCallback;
// Block size can be set in two ways:
// * Through the MLIR lowering pass:
// At this case, blockSizeOverride will be initialized to zero. Then
// the affix tuning parameters pass will decide on a block size.
// Finally, block size will be hinted back to mlir-miopen-driver.
// * Through cmd option "block_size":
// At this case, mlir-miopen-driver assigns a blockSizeOverride. As
// a result, affix tuning parameters pass should make its decisions
// to generate tuning parameters based on this blockSizeOverride.
// This guarantess that affix tuning parameters pass generate
// coherent tuning parameters with the pre-set block size.
int64_t blockSizeOverride;
};
} // anonymous namespace
Expand Down

0 comments on commit ffc7951

Please sign in to comment.