Lines Matching defs:target

42 def testBufferizeToAllocationOpCompact(target):
43     structured.BufferizeToAllocationOp(target)
51 def testBufferizeToAllocationOpArgs(target):
53         target,
70 def testDecompose(target):
71     structured.DecomposeOp(target)
79 def testFuseIntoContainingOpTypes(target):
80     fused = structured.MatchOp.match_op_names(target, ["test.dummy"])
81     containing = structured.MatchOp.match_op_names(target, ["test.dummy"])
95 def testFuseIntoContainingOpCompact(target):
96     fused = structured.MatchOp.match_op_names(target, ["test.dummy"])
97     containing = structured.MatchOp.match_op_names(target, ["test.dummy"])
106 def testFuseOpCompact(target):
108         target, tile_sizes=[4, 8], tile_interchange=[0, 1], apply_cleanup=True
119 def testFuseOpNoArg(target):
120     structured.FuseOp(target)
129 def testFuseOpAttributes(target):
132     structured.FuseOp(target, tile_sizes=attr, tile_interchange=ichange)
142 def testGeneralize(target):
143     structured.GeneralizeOp(target)
151 def testInterchange(target):
152     structured.InterchangeOp(target, iterator_interchange=[1, 0])
161 def testMapCopyToThreadsOpCompact(target):
163         target, total_num_threads=32, desired_bit_alignment=128
174 def testMapCopyToThreadsOpTypes(target):
178         target,
191 def testMatchOpNamesString(target):
192     structured.MatchOp.match_op_names(target, "test.dummy")
201 def testMatchOpNamesList(target):
202     structured.MatchOp.match_op_names(target, ["test.dummy"])
211 def testVectorizeNoArgs(target):
212     structured.VectorizeOp(target)
221 def testVectorizeStatic(target):
222     structured.VectorizeOp(target, [16, 4])
231 def testVectorizeArray(target):
233     structured.VectorizeOp(target, sizes)
242 def testVectorizeMixed(target):
243     sz1 = structured.MatchOp.match_op_names(target, ["arith.constant"])
245     structured.VectorizeOp(target, [sz1, sz2])
255 def testVectorizeEmpty(target):
256     structured.VectorizeOp(target, [])
265 def testVectorizeScalable(target):
266     sz1 = structured.MatchOp.match_op_names(target, ["arith.constant"])
268     structured.VectorizeOp(target, [16, [sz1], [sz2], [8]])
278 def testVectorizeArgs(target):
279     structured.VectorizeOp(target, [16, 4], vectorize_nd_extract=True)
288 def testMatchOpNamesTyped(target):
291         target,
302 def testMultitileSizesCompact(target):
304         transform.AnyOpType.get(), target, dimension=1, target_size=42
319 def testMultitileSizesAllArgs(target):
322         target,
337 def testPadOpNoArgs(target):
338     structured.PadOp(target)
352 def testPadOpArgs(target):
354         target,
375 def testPadOpArgsParam(target):
377         target,
391 def testScalarize(target):
392     structured.ScalarizeOp(target)
399 def testSplit(target):
400     handle = structured.SplitOp(target, dimension=1, chunk_sizes=42)
413 def testTileCompact(target):
414     structured.TileUsingForOp(target, sizes=[4, 8], interchange=[0, 1])
423 def testTileAttributes(target):
426     structured.TileUsingForOp(target, sizes=attr, interchange=ichange)
435 def testTileZero(target):
436     structured.TileUsingForOp(target, sizes=[4, 0, 2, 0], interchange=[0, 1, 2, 3])
467 def testTileExplicitLoopTypeSingle(target):
469         transform.OperationType.get("scf.for"), target, sizes=[2, 3, 4]
478 def testTileExplicitLoopTypeAll(target):
483     structured.TileUsingForOp(types, target, sizes=[2, 3, 4])
492 def testTileScalable(target):
494         target,
504 def testTileToForallCompact(target):
505     matmul = transform.CastOp(transform.OperationType.get("linalg.matmul"), target)
515 def testTileToForallLoopsAndTileOpTypes(target):
519         target,
530 def testTileToForallTileSizes(target):
531     structured.TileUsingForallOp(target, tile_sizes=[2, 3, 4])
539 def testTileToForallMixedDynamic(target):
540     n = structured.MatchOp.match_op_names(target, ["test.dummy"])
541     structured.TileUsingForallOp(target, num_threads=[n, 3, 4])
549 def testTileToForallPackedDynamic(target):
550     n = structured.MatchOp.match_op_names(target, ["test.dummy"])
551     structured.TileUsingForallOp(target, num_threads=n)
559 def testTileToForallMapping(target):
561     structured.TileUsingForallOp(target, num_threads=[2, 3], mapping=mapping)
569 def testVectorizeChildrenAndApplyPatternsAllAttrs(target):
571         target,
588 def testVectorizeChildrenAndApplyPatternsNoAttrs(target):
590         target,
607 def testMatchInterfaceEnum(target):
612         target,
624 def testMatchInterfaceEnumReplaceAttributeBuilder(target):
637         target,