Lines Matching refs:genericOp
54 matchAsPerformingReduction(linalg::GenericOp genericOp);
57 matchAndRewrite(linalg::GenericOp genericOp, OpAdaptor adaptor,
65 linalg::GenericOp genericOp) { in matchAsPerformingReduction() argument
66 Operation *op = genericOp.getOperation(); in matchAsPerformingReduction()
69 if (!genericOp.hasBufferSemantics()) in matchAsPerformingReduction()
73 if (genericOp.getNumInputs() != 1 || genericOp.getNumOutputs() != 1) in matchAsPerformingReduction()
87 if (!genericOp.hasSingleReductionLoop()) in matchAsPerformingReduction()
90 auto indexingMaps = genericOp.getIndexingMapsArray(); in matchAsPerformingReduction()
106 return linalg::RegionMatcher::matchAsScalarBinaryOp(genericOp); in matchAsPerformingReduction()
110 linalg::GenericOp genericOp, OpAdaptor adaptor, in matchAndRewrite() argument
112 Operation *op = genericOp.getOperation(); in matchAndRewrite()
116 auto binaryOpKind = matchAsPerformingReduction(genericOp); in matchAndRewrite()
122 DenseIntElementsAttr localSize = spirv::lookupLocalWorkGroupSize(genericOp); in matchAndRewrite()
137 Location loc = genericOp.getLoc(); in matchAndRewrite()
143 Value x = getLocalInvocationDimSize(genericOp, /*dim=*/0, indexType, loc, in matchAndRewrite()
198 rewriter.eraseOp(genericOp); in matchAndRewrite()