[examples][MLIRPython] reformat python code using black

buddy-compiler · Oct 14, 2023 · c736bb9 · c736bb9
1 parent dd326d5
commit c736bb9
Show file tree

Hide file tree

Showing 5 changed files with 199 additions and 201 deletions.
diff --git a/examples/MLIRPython/.style.yapf b/examples/MLIRPython/.style.yapf
diff --git a/examples/MLIRPython/addmm.py b/examples/MLIRPython/addmm.py
@@ -1,13 +1,13 @@
-from buddy.compiler import DynamoCompiler
+from buddy.compiler import dynamo_compiler
 import torch
 import torch._dynamo as dynamo
 
 
 def foo(c, a, b):
-  return torch.addmm(c, a, b)
+    return torch.addmm(c, a, b)
 
 
-foo_mlir = dynamo.optimize(DynamoCompiler)(foo)
+foo_mlir = dynamo.optimize(dynamo_compiler)(foo)
 
 a_float32 = torch.randn(3, 2)
 b_float32 = torch.randn(2, 3)

diff --git a/examples/MLIRPython/arith_add.py b/examples/MLIRPython/arith_add.py
@@ -4,10 +4,10 @@
 
 
 def foo(x, y):
-  return x + y
+    return x + y
 
 
-foo_mlir = dynamo.optimize(compiler.DynamoCompiler)(foo)
+foo_mlir = dynamo.optimize(compiler.dynamo_compiler)(foo)
 float32_in1 = torch.randn(10).to(torch.float32)
 float32_in2 = torch.randn(10).to(torch.float32)
 foo_mlir(float32_in1, float32_in2)