Skip to content

Commit b49a010

Browse files
committed
compiler: always use long for alloc sizes
1 parent afb0edf commit b49a010

File tree

1 file changed

+9
-13
lines changed

1 file changed

+9
-13
lines changed

devito/passes/iet/definitions.py

Lines changed: 9 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -90,17 +90,13 @@ def __init__(self, rcompile=None, sregistry=None, platform=None, **kwargs):
9090
self.rcompile = rcompile
9191
self.sregistry = sregistry
9292
self.platform = platform
93-
self.index_mode = kwargs.get('options', {'index-mode': 'int32'})['index-mode']
9493

9594
def intm(self, nbytes):
96-
if self.index_mode == 'int64':
97-
try:
98-
syms = retrieve_symbols(nbytes)
99-
return nbytes.subs({s: LONG(s) for s in syms})
100-
except AttributeError:
101-
return LONG(nbytes)
102-
else:
103-
return nbytes
95+
try:
96+
syms = retrieve_symbols(nbytes)
97+
return nbytes.subs({s: LONG(s) for s in syms})
98+
except AttributeError:
99+
return LONG(nbytes)
104100

105101
def _alloc_object_on_low_lat_mem(self, site, obj, storage):
106102
"""
@@ -562,10 +558,10 @@ def process(self, graph):
562558

563559
class DeviceAwareDataManager(DataManager):
564560

565-
def __init__(self, **kwargs):
566-
self.gpu_fit = kwargs['options']['gpu-fit']
567-
self.gpu_create = kwargs['options']['gpu-create']
568-
self.pmode = kwargs['options'].get('place-transfers')
561+
def __init__(self, options=None, **kwargs):
562+
self.gpu_fit = options['gpu-fit']
563+
self.gpu_create = options['gpu-create']
564+
self.pmode = options.get('place-transfers')
569565

570566
super().__init__(**kwargs)
571567

0 commit comments

Comments
 (0)