Skip to content

Commit cc96ce4

Browse files
committed
add to transform
1 parent e80febe commit cc96ce4

File tree

1 file changed

+29
-0
lines changed

1 file changed

+29
-0
lines changed

mlir/extras/dialects/ext/transform.py

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

0 commit comments

Comments
 (0)