Fix input tensors type after merge.

d32bd860 · Adam Osewski · 3caeb7d7 · d32bd860
Commit d32bd860 authored Aug 23, 2023 by Adam Osewski
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdlops_v2r4r2.hpp ...tensor_operation/gpu/grid/gridwise_gemm_xdlops_v2r4r2.hpp +2 -2

No files found.
--- a/include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdlops_v2r4r2.hpp
+++ b/include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdlops_v2r4r2.hpp
@@ -566,8 +566,8 @@ struct GridwiseGemm_bk0mk1_bk0nk1_mn_xdlops_v2r4r2
    template <bool HasMainKBlockLoop,
              InMemoryDataOperationEnum CGlobalMemoryDataOperation,
              typename Block2CTileMap>
-    __device__ static void Run(const FloatAB* p_a_grid,
-                               const FloatAB* p_b_grid,
+    __device__ static void Run(const FloatA* p_a_grid,
+                               const FloatB* p_b_grid,
                               FloatC* p_c_grid,
                               index_t M,
                               index_t N,