Skip to content

Commit d5080db

Browse files
committed
add to transform
1 parent e80febe commit d5080db

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed

mlir/extras/dialects/ext/transform.py

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -469,3 +469,27 @@ def structured_vectorize(
469469
loc=loc,
470470
ip=ip,
471471
)
472+
473+
474+
def structured_vectorize_children_and_apply_patterns(
475+
target,
476+
*,
477+
vectorize_padding=None,
478+
vectorize_nd_extract=None,
479+
flatten_1d_depthwise_conv=None,
480+
disable_multi_reduction_to_contract_patterns=None,
481+
disable_transfer_permutation_map_lowering_patterns=None,
482+
loc=None,
483+
ip=None,
484+
):
485+
return _structured_vectorize_children_and_apply_patterns(
486+
transformed=transform_any_op_t(),
487+
target=target,
488+
vectorize_padding=vectorize_padding,
489+
vectorize_nd_extract=vectorize_nd_extract,
490+
flatten_1d_depthwise_conv=flatten_1d_depthwise_conv,
491+
disable_multi_reduction_to_contract_patterns=disable_multi_reduction_to_contract_patterns,
492+
disable_transfer_permutation_map_lowering_patterns=disable_transfer_permutation_map_lowering_patterns,
493+
loc=loc,
494+
ip=ip,
495+
)

0 commit comments

Comments
 (0)