Lines Matching refs:context
36 TestTileUsingSCFForOpWithFilter(MLIRContext *context, in TestTileUsingSCFForOpWithFilter()
41 : scf::TileUsingSCFForOp(context, options, benefit), filter(filter) {} in TestTileUsingSCFForOpWithFilter()
44 TestTileUsingSCFForOpWithFilter(StringRef opName, MLIRContext *context, in TestTileUsingSCFForOpWithFilter()
49 : scf::TileUsingSCFForOp(context, options, benefit), filter(filter) {} in TestTileUsingSCFForOpWithFilter()
75 MLIRContext *context, scf::SCFTilingOptions options, in TestTileConsumerAndFuseProducersUsingSCFForOpWithFilter()
79 : scf::TileConsumerAndFuseProducersUsingSCFForOp(context, options, in TestTileConsumerAndFuseProducersUsingSCFForOpWithFilter()
85 StringRef opName, MLIRContext *context, scf::SCFTilingOptions options, in TestTileConsumerAndFuseProducersUsingSCFForOpWithFilter()
89 : scf::TileConsumerAndFuseProducersUsingSCFForOp(context, options, in TestTileConsumerAndFuseProducersUsingSCFForOpWithFilter()
144 void addTestPatterns(MLIRContext *context, RewritePatternSet &patterns);
150 addPatternForTiling(MLIRContext *context, RewritePatternSet &patterns, in addPatternForTiling() argument
156 StringAttr::get(context, filterName), StringAttr::get(context, "tiled"));
157 patterns.add<Pattern>(context, tilingOptions, filter);
160 void TestTilingInterfacePass::addTestPatterns(MLIRContext *context, in addTestPatterns() argument
165 context, patterns, "simple_gemm", {10, 20}); in addTestPatterns()
168 context, patterns, "simple_gemm_memref", {10, 20, 30}); in addTestPatterns()
171 context, patterns, "parallel_generic_transpose", {10, 0, 20}); in addTestPatterns()
174 context, patterns, "simple_conv", {0, 0, 0, 0, 10, 20, 30}); in addTestPatterns()
177 context, patterns, "indexed_semantics", {10, 20}); in addTestPatterns()
180 context, patterns, "gemm_interchange", {10, 20, 30}, {1, 2, 0}); in addTestPatterns()
187 context, patterns, "fusion", {10, 20}); in addTestPatterns()
190 context, patterns, "gemm_fusion", {10}); in addTestPatterns()
193 context, patterns, "gemm_interchange_fusion", {10, 20}, {1, 0}); in addTestPatterns()
196 context, patterns, "gemm_plus_gemm_fusion", {10, 20}); in addTestPatterns()
199 context, patterns, "gemm_sequence_fusion", {10}); in addTestPatterns()
205 MLIRContext *context = &getContext(); in runOnOperation() local
207 RewritePatternSet tilingPatterns(context); in runOnOperation()
208 addTestPatterns(context, tilingPatterns); in runOnOperation()