Home
last modified time | relevance | path

Searched refs:vectorSizes (Results 1 – 4 of 4) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Transforms/
H A DSuperVectorize.cpp619 vectorSizes = virtualVectorSize; in Vectorize()
942 return VectorType::get(strategy->vectorSizes, scalarTy); in getVectorType()
1002 assert(state.strategy->vectorSizes.size() == 1 && in createMask()
1056 Type maskTy = VectorType::get(state.strategy->vectorSizes, in createMask()
1291 int64_t forOpVecFactor = strategy.vectorSizes[vectorDim]; in vectorizeAffineForOp()
1657 ArrayRef<int64_t> vectorSizes, in vectorizeLoops() argument
1660 assert((reductionLoops.empty() || vectorSizes.size() == 1) && in vectorizeLoops()
1692 strategy.vectorSizes.assign(vectorSizes.begin(), vectorSizes.end()); in vectorizeLoops()
1725 fastestVaryingPattern.size() != vectorSizes.size()) { in runOnOperation()
1731 if (vectorizeReductions && vectorSizes.size() != 1) { in runOnOperation()
[all …]
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/
H A DUtils.h89 SmallVector<int64_t, 8> vectorSizes; member
115 ArrayRef<int64_t> vectorSizes, ArrayRef<int64_t> fastestVaryingPattern,
H A DPasses.td348 ListOption<"vectorSizes", "virtual-vector-size", "int64_t",
/llvm-project-15.0.7/mlir/test/lib/Dialect/Affine/
H A DTestVectorizationUtils.cpp235 strategy.vectorSizes.push_back(4 /*vectorization factor*/); in testVecAffineLoopNest()