diff options
Diffstat (limited to 'tools/cnn_test.cc')
| -rw-r--r-- | tools/cnn_test.cc | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tools/cnn_test.cc b/tools/cnn_test.cc index d88956d..a209cdf 100644 --- a/tools/cnn_test.cc +++ b/tools/cnn_test.cc @@ -1332,7 +1332,7 @@ int main(int argc, char** argv) { // Create uniform buffers const WGPUBufferDescriptor common_uniform_desc = { .usage = WGPUBufferUsage_Uniform | WGPUBufferUsage_CopyDst, - .size = sizeof(CommonPostProcessUniforms), + .size = sizeof(UniformsSequenceParams), }; WGPUBuffer common_uniform_buffer = wgpuDeviceCreateBuffer(device, &common_uniform_desc); @@ -1384,7 +1384,7 @@ int main(int argc, char** argv) { printf("Processing layer %d/%d...\n", layer + 1, NUM_LAYERS); // Update uniforms - CommonPostProcessUniforms common_u = { + UniformsSequenceParams common_u = { .resolution = {static_cast<float>(width), static_cast<float>(height)}, .aspect_ratio = static_cast<float>(width) / static_cast<float>(height), .time = 0.0f, @@ -1410,7 +1410,7 @@ int main(int argc, char** argv) { BindGroupBuilder() .sampler(0, sampler) .texture(1, current_input) - .buffer(2, common_uniform_buffer, sizeof(CommonPostProcessUniforms)) + .buffer(2, common_uniform_buffer, sizeof(UniformsSequenceParams)) .buffer(3, layer_params_buffer, sizeof(CNNv1LayerParams)) .texture(4, original_view) .build(device, bgl); |