Skip to content

Commit 0ca9e53

Browse files
Update ggml/src/ggml-cuda/conv2d-transpose.cu
Co-authored-by: Aman Gupta <[email protected]>
1 parent ecb82da commit 0ca9e53

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ggml/src/ggml-cuda/conv2d-transpose.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,7 @@ void ggml_cuda_conv_2d_transpose_p0(ggml_backend_cuda_context & ctx, ggml_tensor
127127
GGML_ASSERT(ggml_is_contiguous(kernel));
128128
GGML_ASSERT(ggml_is_contiguous(dst));
129129

130-
const int total = (output_w * output_h * channels_out * batches);
130+
const int total = output_w * output_h * channels_out * batches;
131131
conv2d_transpose_params params = { input_w, input_h, output_w, output_h, kernel_w, kernel_h,
132132
stride, channels_in, channels_out, batches, total };
133133

0 commit comments

Comments
 (0)