1 files changed, 3 insertions, 0 deletions
diff --git a/src/gpu/effects/cnn_v2_effect.h b/src/gpu/effects/cnn_v2_effect.h
index edf301e..facf4c3 100644
--- a/src/gpu/effects/cnn_v2_effect.h
+++ b/src/gpu/effects/cnn_v2_effect.h
@@ -12,6 +12,8 @@ public:
 
   void init(MainSequence* demo) override;
   void resize(int width, int height) override;
+  void compute(WGPUCommandEncoder encoder,
+               const CommonPostProcessUniforms& uniforms) override;
   void render(WGPURenderPassEncoder pass,
               const CommonPostProcessUniforms& uniforms) override;
   void update_bind_group(WGPUTextureView input_view) override;
@@ -36,6 +38,7 @@ private:
   // Input mips
   WGPUTexture input_mip_tex_;
   WGPUTextureView input_mip_view_[3];
+  WGPUTextureView current_input_view_;  // Cached input from update_bind_group
 
   bool initialized_;
 };