From 51028d2f5a2dafb58adbae157b246e904e0eafad Mon Sep 17 00:00:00 2001
From: Emilien Bauer <bauer.emilien@gmail.com>
Date: Tue, 8 Aug 2023 15:36:29 +0100
Subject: [PATCH] Use DeVito's par-tile.

---
 fast/diffusion_2D_wBCs.py | 2 +-
 fast/diffusion_3D_wBCs.py | 2 +-
 fast/wave2d.py            | 5 ++---
 fast/wave3d.py            | 2 +-
 4 files changed, 5 insertions(+), 6 deletions(-)

diff --git a/fast/diffusion_2D_wBCs.py b/fast/diffusion_2D_wBCs.py
index 869d6d2e7d2..0d19b7f733a 100644
--- a/fast/diffusion_2D_wBCs.py
+++ b/fast/diffusion_2D_wBCs.py
@@ -54,7 +54,7 @@
 eq_stencil = Eq(u.forward, stencil)
 
 if args.devito:
-    op = Operator([eq_stencil], name='DevitoOperator')
+    op = Operator([eq_stencil], name='DevitoOperator', opt=('advanced', {'par-tile': (32,4,8)}))
     op.apply(time=nt, dt=dt, a=nu)
     print("Devito Field norm is:", norm(u))
 
diff --git a/fast/diffusion_3D_wBCs.py b/fast/diffusion_3D_wBCs.py
index 9220e25247c..08b248c5ac1 100644
--- a/fast/diffusion_3D_wBCs.py
+++ b/fast/diffusion_3D_wBCs.py
@@ -80,7 +80,7 @@ def plot_3dfunc(u):
 
 # Create Operator
 if args.devito:
-    op = Operator([eq_stencil], name='DevitoOperator')
+    op = Operator([eq_stencil], name='DevitoOperator', opt=('advanced', {'par-tile': (32,4,8)}))
     # Apply the operator for a number of timesteps
     op.apply(time=nt, dt=dt, a=nu)
     print("Devito Field norm is:", norm(u))
diff --git a/fast/wave2d.py b/fast/wave2d.py
index 2a7855a6e95..3c60dc7c3bf 100644
--- a/fast/wave2d.py
+++ b/fast/wave2d.py
@@ -108,7 +108,6 @@
     if args.plot:
         plot_2dfunc(u)
 
-import pdb;pdb.set_trace()
 # print("Init Devito linalg norm 0 :", np.linalg.norm(u.data[0]))
 # print("Init Devito linalg norm 1 :", np.linalg.norm(u.data[1]))
 # print("Init Devito linalg norm 2 :", np.linalg.norm(u.data[2]))
@@ -120,7 +119,7 @@
 
 if args.devito:
     # Run more with no sources now (Not supported in xdsl)
-    op1 = Operator([stencil], name='DevitoOperator')
+    op1 = Operator([stencil], name='DevitoOperator', opt=('advanced', {'par-tile': (32,4,8)}))
     op1.apply(time=time_range.num-1, dt=model.critical_dt)
 
     configuration['mpi'] = 0
@@ -148,7 +147,7 @@
     # print("XDSL init linalg norm:", np.linalg.norm(u.data[2]))
 
     # Run more with no sources now (Not supported in xdsl)
-    xdslop = Operator([stencil], name='xDSLOperator')
+    xdslop = XDSLOperator([stencil], name='xDSLOperator')
     xdslop.apply(time=time_range.num-1, dt=model.critical_dt)
 
     if len(shape) == 2 and args.plot:
diff --git a/fast/wave3d.py b/fast/wave3d.py
index 86b0d55afe3..71602ca30f4 100644
--- a/fast/wave3d.py
+++ b/fast/wave3d.py
@@ -113,7 +113,7 @@
 configuration['mpi'] = mpiconf
 
 # Run more with no sources now (Not supported in xdsl)
-op1 = Operator([stencil], name='DevitoOperator')
+op1 = Operator([stencil], name='DevitoOperator', opt=('advanced', {'par-tile': (32,4,8)}))
 op1.apply(time=time_range.num-1, dt=model.critical_dt)
 
 configuration['mpi'] = 0