Format.

xdslproject · Feb 14, 2024 · 30b57a1 · 30b57a1
1 parent 12eaabe
commit 30b57a1
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 6 deletions.
diff --git a/devito/core/cpu.py b/devito/core/cpu.py
@@ -373,7 +373,10 @@ def _jit_compile(self):
 
                 # xdsl-opt, get xDSL IR
                 # TODO: Remove quotes in pipeline; currently workaround with [1:-1]
-                xdsl_args=[source_name, "--allow-unregistered-dialect", "-p", xdsl_pipeline[1:-1]+','+mlir_pipeline]
+                xdsl_args = [source_name,
+                             "--allow-unregistered-dialect",
+                             "-p",
+                             xdsl_pipeline[1:-1]+','+mlir_pipeline]
                 xdsl = xDSLOptMain(args=xdsl_args)
                 out = io.StringIO()
                 perf("-----------------")
@@ -638,7 +641,10 @@ def _jit_compile(self):
 
                 # xdsl-opt, get xDSL IR
                 # TODO: Remove quotes in pipeline; currently workaround with [1:-1]
-                xdsl_args=[source_name, "--allow-unregistered-dialect", "-p", xdsl_pipeline[1:-1]+','+mlir_pipeline]
+                xdsl_args = [source_name,
+                             "--allow-unregistered-dialect",
+                             "-p",
+                             xdsl_pipeline[1:-1]+','+mlir_pipeline]
                 xdsl = xDSLOptMain(args=xdsl_args)
                 out = io.StringIO()
                 perf("-----------------")
@@ -906,13 +912,19 @@ def generate_XDSL_MPI_PIPELINE(decomp, nb_tiled_dims):
     return generate_pipeline(passes)
 
 
-def generate_pipeline(passes:Iterable[str]):
+def generate_pipeline(passes: Iterable[str]):
     passes_string = ",".join(passes)
     return f'"{passes_string}"'
 
-def generate_mlir_pipeline(passes:Iterable[str]):
+
+def generate_mlir_pipeline(passes: Iterable[str]):
     passes_string = ",".join(passes)
-    return f'mlir-opt{{arguments="--mlir-print-op-generic","--allow-unregistered-dialect","-p","builtin.module({passes_string})"}}'
+    return 'mlir-opt{arguments='\
+           '"--mlir-print-op-generic",'\
+           '"--allow-unregistered-dialect"'\
+           ','\
+           f'"-p","builtin.module({passes_string})"'\
+           '}'
 
 
 # small interop shim script for stuff that we don't want to implement in mlir-ir

diff --git a/devito/core/gpu.py b/devito/core/gpu.py
@@ -439,7 +439,10 @@ def _jit_compile(self):
 
                 # xdsl-opt, get xDSL IR
                 # TODO: Remove quotes in pipeline; currently workaround with [1:-1]
-                xdsl_args=[source_name, "--allow-unregistered-dialect", "-p", xdsl_pipeline[1:-1]+','+mlir_pipeline]
+                xdsl_args = [source_name,
+                             "--allow-unregistered-dialect",
+                             "-p",
+                             xdsl_pipeline[1:-1]+','+mlir_pipeline]
                 xdsl = xDSLOptMain(args=xdsl_args)
                 out = io.StringIO()
                 perf("-----------------")