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,