Searched refs:shuffleOp (Results 1 – 3 of 3) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Conversion/VectorToSPIRV/ |
| H A D | VectorToSPIRV.cpp | 345 matchAndRewrite(vector::ShuffleOp shuffleOp, OpAdaptor adaptor, in matchAndRewrite()
347 auto oldResultType = shuffleOp.getVectorType(); in matchAndRewrite()
352 auto oldSourceType = shuffleOp.getV1VectorType(); in matchAndRewrite()
355 llvm::map_range(shuffleOp.getMask(), [](Attribute attr) -> int32_t { in matchAndRewrite()
359 shuffleOp, newResultType, adaptor.getV1(), adaptor.getV2(), in matchAndRewrite()
367 for (const APInt &i : shuffleOp.getMask().getAsValueRange<IntegerAttr>()) { in matchAndRewrite()
371 shuffleOp, newResultType, newOperands); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/ |
| H A D | AllReduceLowering.cpp | 340 auto shuffleOp = create<gpu::ShuffleOp>( in createSubgroupReduce() local
345 shuffleOp.getResult(1), in createSubgroupReduce()
348 accumFactory(value, shuffleOp.getResult(0))}; in createSubgroupReduce()
362 auto shuffleOp = in createSubgroupReduce() local
365 value = accumFactory(value, shuffleOp.getResult(0)); in createSubgroupReduce()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/VectorToLLVM/ |
| H A D | ConvertVectorToLLVM.cpp | 554 matchAndRewrite(vector::ShuffleOp shuffleOp, OpAdaptor adaptor, in matchAndRewrite() argument
556 auto loc = shuffleOp->getLoc(); in matchAndRewrite()
557 auto v1Type = shuffleOp.getV1VectorType(); in matchAndRewrite()
558 auto v2Type = shuffleOp.getV2VectorType(); in matchAndRewrite()
559 auto vectorType = shuffleOp.getVectorType(); in matchAndRewrite()
561 auto maskArrayAttr = shuffleOp.getMask(); in matchAndRewrite()
578 rewriter.replaceOp(shuffleOp, llvmShuffleOp); in matchAndRewrite()
602 rewriter.replaceOp(shuffleOp, insert); in matchAndRewrite()
|