do not upload gpu data in each frame (cfrenderer)

Author: ChrLackner

ngsolve_webgpu/cf.py

@@ -9,7 +9,7 @@
 from webgpu.colormap import Colormap
 from webgpu.renderer import Renderer, RenderOptions, check_timestamp
 from webgpu.shapes import ShapeRenderer, generate_cylinder
-from webgpu.utils import BufferBinding, UniformBinding, buffer_from_array, write_array_to_buffer, read_shader_file
+from webgpu.utils import BufferBinding, UniformBinding, buffer_from_array, write_array_to_buffer, read_shader_file, device_generation
 from webgpu.renderer import BaseRenderer, RenderOptions, check_timestamp
 from webgpu.uniforms import UniformBase, ct
 from webgpu.utils import (
@@ -212,6 +212,7 @@ class FunctionData:
     order_3d: int
     _timestamp: float = -1
     _gpu_dirty: bool = True
+    _gpu_generation: int = -1
     minval: list[float]
     maxval: list[float]
 
@@ -336,21 +337,24 @@ def get_buffers(self, include_mesh_data=True):
             buffers = self.mesh_data.get_buffers().copy()
         else:
             buffers = {}
+        gen = device_generation()
+        stale = self._gpu_generation != gen
         if self.data_2d is not None:
-            if self._gpu_dirty or self.gpu_2d is None:
+            if self._gpu_dirty or stale or self.gpu_2d is None:
                 self.gpu_2d = buffer_from_array(
                     self.data_2d, label="function_data_2d", reuse=self.gpu_2d
                 )
             buffers["data_2d"] = self.gpu_2d
 
         if self.data_3d is not None:
-            if self._gpu_dirty or self.gpu_3d is None:
+            if self._gpu_dirty or stale or self.gpu_3d is None:
                 self.gpu_3d = buffer_from_array(
                     self.data_3d, label="function_data_3d", reuse=self.gpu_3d
                 )
             buffers["data_3d"] = self.gpu_3d
 
         self._gpu_dirty = False
+        self._gpu_generation = gen
 
         return buffers
 

ngsolve_webgpu/mesh.py

@@ -21,6 +21,7 @@
     BufferBinding,
     UniformBinding,
     buffer_from_array,
+    device_generation,
     get_device,
     read_shader_file,
     uniform_from_array,
@@ -130,6 +131,7 @@ class MeshData:
     _need_3d: bool = False
     _update_lock: Lock
     _gpu_dirty: bool = True
+    _gpu_generation: int = -1
 
     def __init__(self, mesh, el2d_bitarray=None, el3d_bitarray=None):
         import netgen.meshing
@@ -592,6 +594,12 @@ def get_bounding_box(self):
         return ([pmin[0], pmin[1], pmin[2]], [pmax[0], pmax[1], pmax[2]])
 
     def get_buffers(self):
+        gen = device_generation()
+        if self._gpu_generation != gen:
+            self._gpu_dirty = True
+            self.gpu_elements = {}
+            self._dummy_buffer = None
+            self._gpu_generation = gen
 
         if self._gpu_dirty:
             self.gpu_data = buffer_from_array(