| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Interchange.cpp | 59 AffineMap permutationMap = inversePermutation( in interchangeGenericOp() local
61 assert(permutationMap && "unexpected null map"); in interchangeGenericOp()
72 if (!permutationMap.isEmpty()) in interchangeGenericOp()
73 m = m.compose(permutationMap); in interchangeGenericOp()
102 indexOp, permutationMap.getSubMap(indexOp.dim()), allIndices); in interchangeGenericOp()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorTransferPermutationMapRewritePatterns.cpp | 70 AffineMap permutationMap = in matchAndRewrite() local
72 if (permutationMap.isIdentity()) in matchAndRewrite()
75 permutationMap = map.getPermutationMap(permutation, op.getContext()); in matchAndRewrite()
77 permutationMap = inversePermutation(permutationMap); in matchAndRewrite()
78 AffineMap newMap = permutationMap.compose(map); in matchAndRewrite()
|
| H A D | VectorUnrollDistribute.cpp | 107 AffineMap permutationMap, in sliceTransferIndices() argument
118 for (const auto &dim : llvm::enumerate(permutationMap.getResults())) { in sliceTransferIndices()
340 AffineMap permutationMap, in matchAndRewrite()
343 permutationMap, ArrayRef<int64_t>(*targetShape)); in matchAndRewrite()
|
| H A D | VectorTransforms.cpp | 1050 AffineMap permutationMap = AffineMap::getPermutationMap( in matchAndRewrite() local
1053 map = inversePermutation(permutationMap).compose(map); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Transforms/ |
| H A D | SuperVectorize.cpp | 1187 auto permutationMap = makePermutationMap(state.builder.getInsertionBlock(), in vectorizeAffineLoad() local
1189 if (!permutationMap) { in vectorizeAffineLoad()
1194 LLVM_DEBUG(permutationMap.print(dbgs())); in vectorizeAffineLoad()
1197 loadOp.getLoc(), vectorType, loadOp.getMemRef(), indices, permutationMap); in vectorizeAffineLoad()
1232 auto permutationMap = makePermutationMap(state.builder.getInsertionBlock(), in vectorizeAffineStore() local
1234 if (!permutationMap) in vectorizeAffineStore()
1237 LLVM_DEBUG(permutationMap.print(dbgs())); in vectorizeAffineStore()
1241 permutationMap); in vectorizeAffineStore()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/ |
| H A D | VectorOps.cpp | 1604 AffineMap permutationMap) { in build() argument
1607 assert(permutationMap.isProjectedPermutation()); in build()
1612 AffineExpr expr = permutationMap.getResult(i); in build()
2825 for (auto expr : permutationMap.getResults()) { in verifyPermutationMap()
2915 if (permutationMap.getNumSymbols() != 0) in verifyTransferOp()
2926 << AffineMapAttr::get(permutationMap) in verifyTransferOp()
3031 auto permutationMap = getPermutationMap(); in verify() local
3060 return verifyPermutationMap(permutationMap, in verify()
3116 AffineMap permutationMap = op.getPermutationMap(); in foldTransferInBoundsAttribute() local
3489 auto permutationMap = getPermutationMap(); in verify() local
[all …]
|
| /llvm-project-15.0.7/mlir/lib/IR/ |
| H A D | AffineMap.cpp | 213 auto permutationMap = AffineMap::get(*m + 1, 0, affExprs, context); in getPermutationMap() local
214 assert(permutationMap.isPermutation() && "Invalid permutation vector"); in getPermutationMap()
215 return permutationMap; in getPermutationMap()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 2637 AffineMap permutationMap) { in inferTransposeResultType() argument
2642 for (const auto &en : llvm::enumerate(permutationMap.getResults())) in inferTransposeResultType()
2654 map = permutationMap ? map.compose(permutationMap) : map; in inferTransposeResultType()
2663 auto permutationMap = permutation.getValue(); in build() local
2664 assert(permutationMap); in build()
2668 MemRefType resultType = inferTransposeResultType(memRefType, permutationMap); in build()
|
| /llvm-project-15.0.7/mlir/include/mlir/Interfaces/ |
| H A D | VectorInterfaces.td | 220 Helper function to account for the fact that `permutationMap` results and
|
| /llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/ |
| H A D | VectorToGPU.cpp | 360 AffineMap permutationMap = in matchAndRewrite() local
363 permutationMap.compose(transferReadOp.getPermutationMap()); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Vector/IR/ |
| H A D | VectorOps.td | 1342 "AffineMap":$permutationMap,
1477 "AffineMap":$permutationMap,
|