2 files changed, 2 insertions, 0 deletions
diff --git a/src/gpu/effects/cnn_v2_effect.cc b/src/gpu/effects/cnn_v2_effect.cc
index 5e38f13..3985723 100644
--- a/src/gpu/effects/cnn_v2_effect.cc
+++ b/src/gpu/effects/cnn_v2_effect.cc
@@ -530,6 +530,7 @@ void CNNv2Effect::compute(WGPUCommandEncoder encoder,
     params.weight_offset = info.weight_offset;
     params.is_output_layer = (i == layer_info_.size() - 1) ? 1 : 0;
     params.blend_amount = effective_blend;
+    params.is_layer_0 = (i == 0) ? 1 : 0;
 
     wgpuQueueWriteBuffer(ctx_.queue, layer_params_buffers_[i], 0, &params, sizeof(params));
 
diff --git a/src/gpu/effects/cnn_v2_effect.h b/src/gpu/effects/cnn_v2_effect.h
index 47dedf5..8a2e1b6 100644
--- a/src/gpu/effects/cnn_v2_effect.h
+++ b/src/gpu/effects/cnn_v2_effect.h
@@ -45,6 +45,7 @@ private:
     uint32_t weight_offset;
     uint32_t is_output_layer;
     float blend_amount;
+    uint32_t is_layer_0;
   };
 
   struct StaticFeatureParams {