From 30b4f9ce9649cd226bd902381a74017e6241f798 Mon Sep 17 00:00:00 2001 From: Pranathi Vasireddy Date: Wed, 23 Oct 2024 11:50:08 -0600 Subject: [PATCH] Python format --- python/dialects/aie.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/python/dialects/aie.py b/python/dialects/aie.py index f5fab1b860..717f430f92 100644 --- a/python/dialects/aie.py +++ b/python/dialects/aie.py @@ -106,8 +106,12 @@ def __call__(self, *call_args): def bd_dim_layout(size, stride): return Attribute.parse(f"#aie.bd_dim_layout<{size=}, {stride=}>") + def bd_pad_layout(const_pad_before, const_pad_after): - return Attribute.parse(f"#aie.bd_pad_layout<{const_pad_before=}, {const_pad_after=}>") + return Attribute.parse( + f"#aie.bd_pad_layout<{const_pad_before=}, {const_pad_after=}>" + ) + @register_attribute_builder("BDDimLayoutArrayAttr") def bd_dim_layout_array_attr_builder(tups: List[Attribute | Tuple[int]], context=None): @@ -125,7 +129,8 @@ def bd_dim_layout_array_array_attr_builder(tup_arrs: List[List[tuple]], context= f'#aie', context=context, ) - + + @register_attribute_builder("BDPadLayoutArrayAttr") def bd_pad_layout_array_attr_builder( tups: List[Union[Attribute, Tuple[int]]], context=None