misc: Fix use of ExternalAllocator

devitocodes · Sep 12, 2024 · e1a3e43 · e1a3e43
1 parent 94fd903
commit e1a3e43
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/devito/ir/equations/algorithms.py b/devito/ir/equations/algorithms.py
@@ -8,7 +8,7 @@
                           ConditionalDimension)
 from devito.types.basic import AbstractFunction
 from devito.types.dimension import MultiSubDimension
-from devito.data.allocators import ExternalAllocator
+from devito.data.allocators import DataReference
 
 __all__ = ['dimension_sort', 'lower_exprs', 'concretize_subdims']
 
@@ -271,7 +271,7 @@ def _(d, mapper, rebuilt, sregistry):
             rebuilt[idim0] = idim1 = idim0._rebuild(name=iname)
 
         kwargs['implicit_dimension'] = idim1
-        fkwargs.update({'dimensions': (idim1,) + (d.functions.dimensions[1:]),
+        fkwargs.update({'dimensions': (idim1,) + d.functions.dimensions[1:],
                         'halo': None,
                         'padding': None})
 
@@ -280,7 +280,7 @@ def _(d, mapper, rebuilt, sregistry):
     except KeyError:
         fkwargs.update({'name': sregistry.make_name(prefix=d.functions.name),
                         'function': None,
-                        'allocator': ExternalAllocator(d.functions._data)})
+                        'allocator': DataReference(d.functions._data)})
 
         rebuilt[d.functions] = functions = d.functions._rebuild(**fkwargs)