Home
last modified time | relevance | path

Searched refs:isParallelIterator (Results 1 – 8 of 8) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/
H A DNvGpuSupport.cpp296 if (!(isParallelIterator(iteratorTypes[0]) && in matchAndRewrite()
297 isParallelIterator(iteratorTypes[1]) && in matchAndRewrite()
H A DVectorToGPU.cpp73 if (!(isParallelIterator(iteratorTypes[0]) && in contractSupportsMMAMatrixType()
74 isParallelIterator(iteratorTypes[1]) && in contractSupportsMMAMatrixType()
295 if (!(isParallelIterator(iteratorTypes[0]) && in matchAndRewrite()
296 isParallelIterator(iteratorTypes[1]) && in matchAndRewrite()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Utils/
H A DStructuredOpsUtils.h81 inline bool isParallelIterator(Attribute attr) { in isParallelIterator() function
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/
H A DUtils.cpp516 if (isParallelIterator(iteratorType.value())) in doit()
548 if (isParallelIterator(iteratorType.value())) in doit()
630 nLoops - iteratorTypes.drop_while(isParallelIterator).size();
757 if (isParallelIterator(iteratorType.value())) in doit()
768 if (isParallelIterator(iteratorType.value())) { in doit()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DFusionOnTensors.cpp441 return !isParallelIterator(iterType); in tileConsumerAndFuseProducers()
H A DElementwiseOpFusion.cpp1048 if (!isParallelIterator(startIteratorType) && in getCollapsableIterationSpaceDims()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorDropLeadUnitDim.cpp312 if (!isParallelIterator(oldIteratorTypes[dimToDrop])) in matchAndRewrite()
H A DVectorTransforms.cpp1365 if (!isParallelIterator(iteratorTypes[0]) || in matchAndRewrite()
1366 !isParallelIterator(iteratorTypes[1]) || in matchAndRewrite()
1667 if (isParallelIterator(iteratorTypes[0]) && in matchAndRewrite()
1668 isParallelIterator(iteratorTypes[1]) && in matchAndRewrite()
1700 } else if (isParallelIterator(iteratorTypes[0]) && in matchAndRewrite()