Update reference calculation

8038fad9 · Rostyslav Geyyer · 7cf5d8f7 · 8038fad9
Commit 8038fad9 authored Sep 20, 2023 by Rostyslav Geyyer
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 3 deletions

library/include/ck/library/reference_tensor_operation/cpu/reference_gemm.hpp ...library/reference_tensor_operation/cpu/reference_gemm.hpp +4 -3

No files found.
--- a/library/include/ck/library/reference_tensor_operation/cpu/reference_gemm.hpp
+++ b/library/include/ck/library/reference_tensor_operation/cpu/reference_gemm.hpp
@@ -21,7 +21,8 @@ template <typename ADataType,
          typename AElementwiseOperation,
          typename BElementwiseOperation,
          typename CElementwiseOperation,
-          typename ComputType = ADataType>
+          typename ComputeTypeA = ADataType,
+          typename ComputeTypeB = ComputeTypeA>
 struct ReferenceGemm : public device::BaseOperator
 {
    // Argument
@@ -65,8 +66,8 @@ struct ReferenceGemm : public device::BaseOperator

                for(int k = 0; k < K; ++k)
                {
-                    ComputType v_a;
-                    ComputType v_b;
+                    ComputeTypeA v_a;
+                    ComputeTypeB v_b;

                    // use PassThrough instead of ConvertBF16RTN for reference calculation
                    if constexpr(is_same_v<AElementwiseOperation,