Merge remote-tracking branch 'origin/develop' into gelu

09ec28be · Chao Liu · b9d3d277 · 85fc91c3 · 09ec28be
Commit 09ec28be authored May 31, 2022 by Chao Liu
Show 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();