From 4a2b155d577f6d6ad49399a3af5f829148733fba Mon Sep 17 00:00:00 2001
From: George Bisbas <g.bisbas18@imperial.ac.uk>
Date: Thu, 23 Nov 2023 18:36:32 +0000
Subject: [PATCH] compiler: Rework correctness for indirections

---
 devito/ir/clusters/cluster.py   | 7 ++++++-
 devito/types/dimension.py       | 2 +-
 examples/userapi/02_apply.ipynb | 4 ++--
 3 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/devito/ir/clusters/cluster.py b/devito/ir/clusters/cluster.py
index 2620e88e0ae..74754bad96f 100644
--- a/devito/ir/clusters/cluster.py
+++ b/devito/ir/clusters/cluster.py
@@ -388,7 +388,12 @@ def dspace(self):
         for f, v in parts.items():
             for i in v:
                 if i.dim in oobs:
-                    intervals = intervals.ceil(v[i.dim])
+                    try:
+                        if intervals[i.dim].upper > v[i.dim].upper and \
+                           bool(i.dim in f.dimensions):
+                            intervals = intervals.ceil(v[i.dim])
+                    except AttributeError:
+                        pass
 
         return DataSpace(intervals, parts)
 
diff --git a/devito/types/dimension.py b/devito/types/dimension.py
index 88ec09d703c..f9a6847a9e5 100644
--- a/devito/types/dimension.py
+++ b/devito/types/dimension.py
@@ -342,7 +342,7 @@ def _arg_check(self, args, size, interval):
                 # Autopadding causes non-integer upper limit
                 from devito.symbolics import normalize_args
                 upper = interval.upper.subs(normalize_args(args))
-            if args[self.max_name] + upper > size:
+            if args[self.max_name] + upper >= size:
                 raise InvalidArgument("OOB detected due to %s=%d" % (self.max_name,
                                                                      args[self.max_name]))
 
diff --git a/examples/userapi/02_apply.ipynb b/examples/userapi/02_apply.ipynb
index 80584051bda..f8f730dd522 100644
--- a/examples/userapi/02_apply.ipynb
+++ b/examples/userapi/02_apply.ipynb
@@ -250,14 +250,14 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "OOB detected due to time_M=3\n"
+      "OOB detected due to time_M=2\n"
      ]
     }
    ],
    "source": [
     "from devito.exceptions import InvalidArgument\n",
     "try:\n",
-    "    op.apply(time_M=3)\n",
+    "    op.apply(time_M=2)\n",
     "except InvalidArgument as e:\n",
     "    print(e)"
    ]