Merge remote-tracking branch 'origin/develop' into myamlak/cgemm

bda26547 · root · 80f038a0 · 3e6c2610 · bda26547
Commit bda26547 authored May 27, 2022 by root
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 2 deletions

test/grouped_gemm/grouped_gemm_fp16.cpp test/grouped_gemm/grouped_gemm_fp16.cpp +7 -2

No files found.
--- a/test/grouped_gemm/grouped_gemm_fp16.cpp
+++ b/test/grouped_gemm/grouped_gemm_fp16.cpp
@@ -151,8 +151,13 @@ bool TestGroupedGemm(DeviceGroupedGemmPtr_& groupedGemmPtr)
    {
        c_tensors_device[i]->FromDevice(c_device_tensors[i].mData.data());

-        using ReferenceGemmInstance = ck::tensor_operation::host::
-            ReferenceGemm<ADataType, BDataType, CDataType, PassThrough, PassThrough, PassThrough>;
+        using ReferenceGemmInstance = ck::tensor_operation::host::ReferenceGemm<ADataType,
+                                                                                BDataType,
+                                                                                CDataType,
+                                                                                AccDataType,
+                                                                                PassThrough,
+                                                                                PassThrough,
+                                                                                PassThrough>;

        auto ref_gemm    = ReferenceGemmInstance{};
        auto ref_invoker = ref_gemm.MakeInvoker();