Searched refs:interchangeVector (Results 1 – 8 of 8) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Interchange.cpp | 38 ArrayRef<unsigned> interchangeVector) { in interchangeGenericOpPrecondition() argument
40 if (interchangeVector.empty() || in interchangeGenericOpPrecondition()
41 genericOp.getNumLoops() != interchangeVector.size()) in interchangeGenericOpPrecondition()
44 if (!inversePermutation(AffineMap::getPermutationMap(interchangeVector, in interchangeGenericOpPrecondition()
52 ArrayRef<unsigned> interchangeVector) { in interchangeGenericOp() argument
53 if (failed(interchangeGenericOpPrecondition(genericOp, interchangeVector))) in interchangeGenericOp()
60 AffineMap::getPermutationMap(interchangeVector, context)); in interchangeGenericOp()
83 SmallVector<int64_t> permutation(interchangeVector.begin(), in interchangeGenericOp()
84 interchangeVector.end()); in interchangeGenericOp()
|
| H A D | Tiling.cpp | 419 if (!options.interchangeVector.empty()) { in tileLinalgOpImpl()
422 SmallVector<unsigned, 4> interchangeVector; in tileLinalgOpImpl() local
423 interchangeVector.reserve(options.interchangeVector.size()); in tileLinalgOpImpl()
424 for (auto pos : options.interchangeVector) { in tileLinalgOpImpl()
428 interchangeVector.push_back(it->second); in tileLinalgOpImpl()
433 AffineMap::getPermutationMap(interchangeVector, b.getContext())); in tileLinalgOpImpl()
435 SmallVector<int64_t> permutation(interchangeVector.begin(), in tileLinalgOpImpl()
436 interchangeVector.end()); in tileLinalgOpImpl()
453 if (!options.interchangeVector.empty()) in tileLinalgOpImpl()
|
| H A D | Transforms.cpp | 536 MLIRContext *context, ArrayRef<unsigned> interchangeVector, in GenericOpInterchangePattern() argument
539 interchangeVector(interchangeVector.begin(), interchangeVector.end()) {} in GenericOpInterchangePattern()
548 interchangeGenericOp(rewriter, genericOp, interchangeVector); in returningMatchAndRewrite()
|
| H A D | LinalgStrategyPasses.cpp | 219 SmallVector<unsigned> interchangeVector(iteratorInterchange.begin(), in runOnOperation() local
223 funcOp.getContext(), interchangeVector, filter); in runOnOperation()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/ |
| H A D | TileUsingInterface.cpp | 48 fillInterchangeVector(ArrayRef<unsigned> interchangeVector, in fillInterchangeVector() argument
50 SmallVector<unsigned> filledVector = llvm::to_vector(interchangeVector); in fillInterchangeVector()
195 SmallVector<unsigned> interchangeVector; in returningMatchAndRewrite() local
196 if (!options.interchangeVector.empty()) { in returningMatchAndRewrite()
197 interchangeVector = fillInterchangeVector(options.interchangeVector, in returningMatchAndRewrite()
200 if (!interchangeVector.empty()) { in returningMatchAndRewrite()
201 if (!isPermutation(interchangeVector)) { in returningMatchAndRewrite()
208 applyPermutationToVector(iterationDomain, interchangeVector); in returningMatchAndRewrite()
210 applyPermutationToVector(tileSizeVector, interchangeVector); in returningMatchAndRewrite()
219 if (!interchangeVector.empty()) { in returningMatchAndRewrite()
[all …]
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/SCF/Transforms/ |
| H A D | TileUsingInterface.h | 56 SmallVector<unsigned> interchangeVector = {}; member
58 interchangeVector = llvm::to_vector(interchange); in setInterchange()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Transforms/ |
| H A D | Transforms.h | 185 ArrayRef<unsigned> interchangeVector);
618 SmallVector<unsigned, 4> interchangeVector = {}; member
621 interchangeVector.assign(interchange.begin(), interchange.end()); in setInterchange()
839 MLIRContext *context, ArrayRef<unsigned> interchangeVector,
857 SmallVector<unsigned, 8> interchangeVector; member
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/TransformOps/ |
| H A D | LinalgTransformOps.cpp | 401 SmallVector<unsigned> interchangeVector = in applyToOne() local
404 if (interchangeVector.empty()) { in applyToOne()
410 interchangeGenericOp(rewriter, target, interchangeVector); in applyToOne()
|