Reference contraction: Add missing type_convert for dst value

a1d9285f · Bartlomiej Wroblewski · 0ceab3dd · a1d9285f
Commit a1d9285f authored Sep 22, 2023 by Bartlomiej Wroblewski
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

library/include/ck/library/reference_tensor_operation/cpu/reference_contraction.hpp .../reference_tensor_operation/cpu/reference_contraction.hpp +1 -1

No files found.
--- a/library/include/ck/library/reference_tensor_operation/cpu/reference_contraction.hpp
+++ b/library/include/ck/library/reference_tensor_operation/cpu/reference_contraction.hpp
@@ -81,7 +81,7 @@ struct ReferenceContraction_M2_N2_K2 : public ck::tensor_operation::device::Base
                    }
                }

-                arg.c_ms_ns_(m0, m1, n0, n1) = v_acc;
+                arg.c_ms_ns_(m0, m1, n0, n1) = ck::type_convert<CDataType>(v_acc);
            };

            make_ParallelTensorFunctor(f_ms_ns,