summaryrefslogtreecommitdiff
path: root/src/tests/gpu
diff options
context:
space:
mode:
Diffstat (limited to 'src/tests/gpu')
-rw-r--r--src/tests/gpu/test_cnn_v3_parity.cc4
1 files changed, 2 insertions, 2 deletions
diff --git a/src/tests/gpu/test_cnn_v3_parity.cc b/src/tests/gpu/test_cnn_v3_parity.cc
index 15fe818..1a7f169 100644
--- a/src/tests/gpu/test_cnn_v3_parity.cc
+++ b/src/tests/gpu/test_cnn_v3_parity.cc
@@ -190,8 +190,8 @@ static std::vector<float> run_cnn_v3(WebGPUTestFixture& fixture,
effect.upload_weights(ctx.queue, weights_u32, weights_bytes);
} else {
// Explicitly zero weights to override any asset-loaded defaults.
- // kWeightsBufBytes = ((1964+1)/2)*4 = 3928
- const uint32_t zero_size = ((1964u + 1u) / 2u) * 4u;
+ // kWeightsBufBytes = ((2476+1)/2)*4 = 4952
+ const uint32_t zero_size = ((2476u + 1u) / 2u) * 4u;
std::vector<uint8_t> zeros(zero_size, 0);
effect.upload_weights(ctx.queue, zeros.data(), zero_size);
}