Lines Matching defs:ny
53 /// <namePrefix><xPerm>_<x type>_coo_<ny>_<y0 type>..._<yn type> for sort_coo
56 uint64_t ny, ValueRange operands) {
62 nameOstream << "_coo_" << ny;
71 /// parameters `xPerm` and `ny` tell the number of x and y values provided
80 StringRef namePrefix, AffineMap xPerm, uint64_t ny, ValueRange operands,
84 getMangledSortHelperFuncName(nameOstream, namePrefix, xPerm, ny,
101 createFunc(builder, module, func, xPerm, ny, nTrailingP);
111 uint64_t ny,
113 Value cstep = constantIndex(builder, loc, xPerm.getNumResults() + ny);
131 uint64_t ny,
134 // Create code for the first (xPerm + ny) buffers.
136 for (unsigned y = 0; y < ny; y++) {
150 bodyBuilder(arg.index() + xPerm.getNumResults() + ny, i, j, arg.value());
166 AffineMap xPerm, uint64_t ny) {
174 forEachIJPairInAllBuffers(builder, loc, args, xPerm, ny, swapOnePair);
181 uint64_t ny,
200 forEachIJPairInXs(builder, loc, args, xPerm, ny, bodyBuilder);
251 uint64_t ny, uint32_t nTrailingP = 0) {
255 return createInlinedCompareImplementation(builder, loc, args, xPerm, ny,
305 uint64_t ny, uint32_t nTrailingP = 0) {
309 return createInlinedCompareImplementation(builder, loc, args, xPerm, ny,
328 uint64_t ny, uint32_t nTrailingP = 0) {
369 Value cond2 = createInlinedLessThan(builder, loc, compareOperands, xPerm, ny);
393 uint64_t ny, int step) {
411 Value cond = createInlinedLessThan(builder, loc, compareOperands, xPerm, ny);
426 createInlinedEqCompare(builder, loc, compareOperands, xPerm, ny);
435 AffineMap xPerm, uint64_t ny,
442 Value cond = createInlinedLessThan(builder, loc, compareOperands, xPerm, ny);
447 createSwap(builder, loc, swapOperands, xPerm, ny);
453 uint64_t ny, SmallVectorImpl<Value> &swapOperands,
456 scf::IfOp ifOp = createCompareThenSwap(builder, loc, xPerm, ny, swapOperands,
458 createCompareThenSwap(builder, loc, xPerm, ny, swapOperands, compareOperands,
465 uint64_t ny, SmallVectorImpl<Value> &swapOperands,
469 scf::IfOp ifOp1 = createCompareThenSwap(builder, loc, xPerm, ny, swapOperands,
474 createInsert3rd(builder, loc, xPerm, ny, swapOperands, compareOperands, v0,
480 uint64_t ny, SmallVectorImpl<Value> &swapOperands,
484 createSort3(builder, loc, xPerm, ny, swapOperands, compareOperands, v0, v1,
489 builder, loc, xPerm, ny, swapOperands, compareOperands, v2, v3);
490 createInsert3rd(builder, loc, xPerm, ny, swapOperands, compareOperands, v0,
499 scf::IfOp ifOp = createCompareThenSwap(builder, loc, xPerm, ny, swapOperands,
510 func::FuncOp func, AffineMap xPerm, uint64_t ny,
529 createSort3(builder, loc, xPerm, ny, swapOperands, compareOperands, lo, mi,
541 createSort5(builder, loc, xPerm, ny, swapOperands, compareOperands, lo, a, mi,
578 func::FuncOp func, AffineMap xPerm, uint64_t ny,
598 createChoosePivot(builder, module, func, xPerm, ny, i, j, p, args);
623 i, p, xPerm, ny, 1);
627 j, p, xPerm, ny, -1);
637 createSwap(builder, loc, swapOperands, xPerm, ny);
729 func::FuncOp func, AffineMap xPerm, uint64_t ny,
786 createInlinedLessThan(builder, loc, compareOperands, xPerm, ny);
817 Value cond = createInlinedLessThan(builder, loc, compareOperands, xPerm, ny);
827 createSwap(builder, loc, swapOperands, xPerm, ny);
860 func::FuncOp func, AffineMap xPerm, uint64_t ny,
888 builder, func, TypeRange(), kShiftDownFuncNamePrefix, xPerm, ny,
903 createSwap(builder, loc, swapOperands, xPerm, ny);
919 ValueRange args, AffineMap xPerm, uint64_t ny,
929 ny, args.drop_back(nTrailingP), createPartitionFunc);
1000 uint64_t ny, uint32_t nTrailingP) {
1026 xPerm, ny, operands, createBinarySearchFunc);
1036 builder, loc, operands, xPerm, ny,
1052 builder, loc, operands, xPerm, ny,
1062 builder, loc, operands, xPerm, ny,
1114 func::FuncOp func, AffineMap xPerm, uint64_t ny,
1164 builder, func, TypeRange(), kSortStableFuncNamePrefix, xPerm, ny,
1184 builder, func, TypeRange(), kHeapSortFuncNamePrefix, xPerm, ny,
1194 createQuickSort(builder, module, func, args, xPerm, ny, nTrailingP);
1207 createQuickSort(builder, module, func, args, xPerm, ny, nTrailingP);
1221 uint64_t ny, PatternRewriter &rewriter) {
1276 xPerm, ny, operands, funcGenerator, nTrailingP);
1411 uint64_t ny = 0;
1413 ny = nyAttr.getInt();
1415 return matchAndRewriteSortOp(op, xys, xPerm, ny, rewriter);