1 files changed, 4 insertions, 0 deletions
diff --git a/cnn_v3/src/cnn_v3_effect.h b/cnn_v3/src/cnn_v3_effect.h
index c358990..36e2797 100644
--- a/cnn_v3/src/cnn_v3_effect.h
+++ b/cnn_v3/src/cnn_v3_effect.h
@@ -89,6 +89,10 @@ class CNNv3Effect : public Effect {
   // Update FiLM conditioning; call before render() each frame.
   void set_film_params(const CNNv3FiLMParams& fp);
 
+  // Upload packed-f16 weights (kWeightsBufBytes bytes of u32 pairs).
+  // Used for testing and inference from trained .bin files.
+  void upload_weights(WGPUQueue queue, const void* data, uint32_t size_bytes);
+
  private:
   // Intermediate node names (prefixed from output[0])
   std::string node_enc0_;