debug memory leak in wave

Author: matthiasdiener

examples/wave.py

@@ -124,10 +124,10 @@ def main(actx_class, use_profiling=False, use_logmgr=False, lazy: bool = False):
         logmgr_add_device_memory_usage(logmgr, queue)
         logmgr_add_mempool_usage(logmgr, alloc)
 
-        logmgr.add_watches(["step.max", "t_step.max", "t_log.max"])
+        logmgr.add_watches(["step.max"])
 
         try:
-            logmgr.add_watches(["memory_usage_python.max", "memory_usage_gpu.max"])
+            logmgr.add_watches(["memory_usage_mempool_managed.max", "memory_usage_mempool_active.max"])
         except KeyError:
             pass
 
@@ -152,18 +152,19 @@ def rhs(t, w):
         if logmgr:
             logmgr.tick_before()
 
+        gc.collect()
+
         fields = rk4_step(fields, t, dt, compiled_rhs)
         fields = force_evaluation(actx, fields)
 
-        if istep % 10 == 0:
-            if use_profiling:
-                print(actx.tabulate_profiling_data())
-            print(istep, t, actx.to_numpy(op.norm(dcoll, fields[0], 2)))
-            vis.write_vtk_file("fld-wave-%04d.vtu" % istep,
-                    [
-                        ("u", fields[0]),
-                        ("v", fields[1:]),
-                        ], overwrite=True)
+        if istep == 10:
+            gc.set_debug(gc.DEBUG_SAVEALL)
+            gc.collect()
+            gc.set_debug(0)
+            obj_list = gc.garbage
+
+            for obj in obj_list:
+                print(type(obj))
 
         t += dt
         istep += 1