Fix invalid in_element op in ref_conv

5ca4b157 · rocking · af9b4f25 · 5ca4b157
Commit 5ca4b157 authored Jun 27, 2023 by rocking
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

library/include/ck/library/reference_tensor_operation/cpu/reference_conv_bwd_data.hpp ...eference_tensor_operation/cpu/reference_conv_bwd_data.hpp +3 -3

No files found.
--- a/library/include/ck/library/reference_tensor_operation/cpu/reference_conv_bwd_data.hpp
+++ b/library/include/ck/library/reference_tensor_operation/cpu/reference_conv_bwd_data.hpp
@@ -125,7 +125,7 @@ struct ReferenceConvBwdData : public device::BaseOperator
                    arg.in_element_op_(v_in, v_acc);
-                    arg.input_(g, n, c, wi) = ck::type_convert<InDataType>(v_acc);
+                    arg.input_(g, n, c, wi) = ck::type_convert<InDataType>(v_in);
                };
                make_ParallelTensorFunctor(f_ncw,
@@ -201,7 +201,7 @@ struct ReferenceConvBwdData : public device::BaseOperator
                    arg.in_element_op_(v_in, v_acc);
-                    arg.input_(g, n, c, hi, wi) = ck::type_convert<InDataType>(v_acc);
+                    arg.input_(g, n, c, hi, wi) = ck::type_convert<InDataType>(v_in);
                };
                make_ParallelTensorFunctor(f_nchw,
@@ -299,7 +299,7 @@ struct ReferenceConvBwdData : public device::BaseOperator
                    arg.in_element_op_(v_in, v_acc);
-                    arg.input_(g, n, c, di, hi, wi) = ck::type_convert<InDataType>(v_acc);
+                    arg.input_(g, n, c, di, hi, wi) = ck::type_convert<InDataType>(v_in);
                };
                make_ParallelTensorFunctor(f_ncdhw,