Skip to content

Commit 2c63e42

Browse files
committed
In aie.extras ShapedValue changed from a Class to a function
1 parent 740a6a1 commit 2c63e42

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

python/dialects/aie.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -257,7 +257,8 @@ def __init__(self, tile, link_with=None, dynamic_objfifo_lowering=None):
257257
# Create an aie buffer of (shape x datatype) on given tile.
258258
# shape examples: [256], [256, 256], [256, 256,]
259259
# This class hides the BufferOp and instead pretends to be a MemRef
260-
class buffer(BufferOp, ShapedValue):
260+
@ShapedValue
261+
class buffer(BufferOp):
261262
def __init__(self):
262263
raise ValueError("Should never be called")
263264

0 commit comments

Comments
 (0)