From 5aba133d924a471a027949bb80cfc44bb612c4c2 Mon Sep 17 00:00:00 2001
From: rgarcia <rgarcia@iasolver.com>
Date: Tue, 16 Jul 2024 09:18:52 +0200
Subject: [PATCH] Fix typo in previous commit

---
 src/descriptors/descriptor_convT2D.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/descriptors/descriptor_convT2D.cpp b/src/descriptors/descriptor_convT2D.cpp
index 3c13c221..a0aad430 100644
--- 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);