Searched refs:broadcastOp (Results 1 – 3 of 3) sorted by relevance
32 matchAndRewrite(vector::BroadcastOp broadcastOp, ArrayRef<Value> operands, in matchAndRewrite()34 if (broadcastOp.source().getType().isa<VectorType>() || in matchAndRewrite()35 !spirv::CompositeType::isValid(broadcastOp.getVectorType())) in matchAndRewrite()38 SmallVector<Value, 4> source(broadcastOp.getVectorType().getNumElements(), in matchAndRewrite()41 broadcastOp.getLoc(), broadcastOp.getVectorType(), source); in matchAndRewrite()42 rewriter.replaceOp(broadcastOp, construct); in matchAndRewrite()
2028 static LogicalResult verify(spirv::GroupBroadcastOp broadcastOp) { in verify() argument2029 spirv::Scope scope = broadcastOp.execution_scope(); in verify()2031 return broadcastOp.emitOpError( in verify()2034 if (auto localIdTy = broadcastOp.localid().getType().dyn_cast<VectorType>()) in verify()2036 return broadcastOp.emitOpError("localid is a vector and can be with only " in verify()2060 static LogicalResult verify(spirv::GroupNonUniformBroadcastOp broadcastOp) { in verify() argument2061 spirv::Scope scope = broadcastOp.execution_scope(); in verify()2063 return broadcastOp.emitOpError( in verify()2068 auto targetEnv = spirv::getDefaultTargetEnv(broadcastOp.getContext()); in verify()2069 if (auto spirvModule = broadcastOp->getParentOfType<spirv::ModuleOp>()) in verify()[all …]
816 auto broadcastOp = extractOp.vector().getDefiningOp<vector::BroadcastOp>(); in foldExtractFromBroadcast() local817 if (!broadcastOp) in foldExtractFromBroadcast()819 if (extractOp.getType() == broadcastOp.getSourceType()) in foldExtractFromBroadcast()820 return broadcastOp.source(); in foldExtractFromBroadcast()824 unsigned broadcasrSrcRank = getRank(broadcastOp.getSourceType()); in foldExtractFromBroadcast()832 extractOp.setOperand(broadcastOp.source()); in foldExtractFromBroadcast()