Fix typo in previous commit

deephealthproject · Jul 16, 2024 · 5aba133 · 5aba133
1 parent fa99e51
commit 5aba133
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/descriptors/descriptor_convT2D.cpp b/src/descriptors/descriptor_convT2D.cpp
@@ -214,7 +214,7 @@ void ConvolDescriptorT2D::build(Tensor *A) {
                  in,iz,ir,ic);
    cudnnCreateFilterDescriptor(&wDesc);
    //CONVT we need to swap input channels with output so all other swappings (forward and backward functions) matches
-   cudnnSetFilter4dDescriptor(wDesc, data_type, tensor_format, kz, nz, kr, kc);
+   cudnnSetFilter4dDescriptor(wDesc, data_type, tensor_format, kz, nk, kr, kc);
 
    cudnnCreateTensorDescriptor(&yDesc);
    cudnnSetTensor4dDescriptor(yDesc, tensor_format, data_type, in, z,r,c);