Ensure wavefield reallocation with nvidia

trustimaging · Oct 5, 2023 · b7a6285 · b7a6285
1 parent 2779afc
commit b7a6285
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/stride/physics/iso_acoustic/devito.py b/stride/physics/iso_acoustic/devito.py
@@ -336,7 +336,7 @@ async def before_forward(self, wavelets, vp, rho=None, alpha=None, **kwargs):
 
         else:
             # If the wavefield is lazily streamed, re-create every time
-            if save_wavefield and 'nvidia' in platform and devito.pro_available:
+            if 'nvidia' in platform and devito.pro_available:
                 self.dev_grid.undersampled_time_function('p_saved',
                                                          bounds=kwargs.pop('save_bounds', None),
                                                          factor=self.undersampling_factor,