intel/llvm
1
0
Fork 0
mirror of https://github.com/intel/llvm.git synced 2026-02-08 17:28:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Set the namespace of the StandardOps dialect to "std", but add a special case to the parser to allow parsing standard operations without the "std" prefix. This will now allow for the standard dialect to be looked up dynamically by name.

Browse Source 
PiperOrigin-RevId: 236493865
This commit is contained in:
River Riddle
2019-03-02 18:03:03 -08:00
committed by jpienaar
parent eee85361bb
commit eeeef090ef
18 changed files with 309 additions and 258 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
mlir/bindings/python/test/test_py2and3.py
View File

@@ -34,10 +34,10 @@ class EdscTest(unittest.TestCase):
with E.ContextManager():
a, b = (E.Expr(E.Bindable(self.i32Type)) for _ in range(2))
c1 = self.module.op(
"constant",
"std.constant",
self.i32Type, [],
value=self.module.integerAttr(self.i32Type, 42))
expr = self.module.op("addi", self.i32Type, [c1, b])
expr = self.module.op("std.addi", self.i32Type, [c1, b])
str = expr.__str__()
self.assertIn("addi(42, $2)", str)
@@ -99,7 +99,7 @@ class EdscTest(unittest.TestCase):
A, B, C = list(map(E.Indexed, [E.Bindable(memrefType) for _ in range(3)]))
stmt = C.store([i, j], A.load([i, k]) * B.load([k, j]))
str = stmt.__str__()
self.assertIn(" = store(", str)
self.assertIn(" = std.store(", str)
def testMatmul(self):
with E.ContextManager():
@@ -118,7 +118,7 @@ class EdscTest(unittest.TestCase):
self.assertIn("for($1 = $4 to $7 step $10) {", str)
self.assertIn("for($2 = $5 to $8 step $11) {", str)
self.assertIn("for($3 = $6 to $9 step $12) {", str)
self.assertIn(" = store", str)
self.assertIn(" = std.store", str)
def testArithmetic(self):
with E.ContextManager():
@@ -353,10 +353,10 @@ class EdscTest(unittest.TestCase):
emitter = E.MLIRFunctionEmitter(f)
funcArg, = emitter.bind_function_arguments()
c1 = self.module.op(
"constant",
"std.constant",
self.i32Type, [],
value=self.module.integerAttr(self.i32Type, 42))
expr = self.module.op("addi", self.i32Type, [c1, funcArg])
expr = self.module.op("std.addi", self.i32Type, [c1, funcArg])
block = E.Block([E.Stmt(expr), E.Return()])
emitter.emit_inplace(block)
self.module.compile()