Magic-Mirror/llvm-capstone
1
0
Fork 0
mirror of https://github.com/capstone-engine/llvm-capstone.git synced 2024-11-30 09:01:19 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[mlir][python] add MemRefTypeAttr attr builder (#76371)

Browse Source
This commit is contained in:
Maksim Levental 2024-01-06 16:42:14 -06:00 committed by GitHub
parent 6e1ecd100d
commit 83be8a7400
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mlir/python/mlir/ir.py
View File

@ -263,6 +263,11 @@ def _typeArrayAttr(x, context):
return _arrayAttr([TypeAttr.get(t, context=context) for t in x], context)
@register_attribute_builder("MemRefTypeAttr")
def _memref_type_attr(x, context):
return _typeAttr(x, context)
try:
import numpy as np

12
mlir/test/python/dialects/memref.py
View File

@ -3,6 +3,7 @@
from mlir.ir import *
import mlir.dialects.func as func
import mlir.dialects.memref as memref
import mlir.extras.types as T
def run(f):
@ -76,3 +77,14 @@ def testCustomBuidlers():
# CHECK: memref.load %[[ARG0]][%[[ARG1]], %[[ARG2]]]
print(module)
assert module.operation.verify()
# CHECK-LABEL: TEST: testMemRefAttr
@run
def testMemRefAttr():
with Context() as ctx, Location.unknown(ctx):
module = Module.create()
with InsertionPoint(module.body):
memref.global_("objFifo_in0", T.memref(16, T.i32()))
# CHECK: memref.global @objFifo_in0 : memref<16xi32>
print(module)