Commit d32bd860 authored by Adam Osewski's avatar Adam Osewski
Browse files

Fix input tensors type after merge.

parent 3caeb7d7
...@@ -566,8 +566,8 @@ struct GridwiseGemm_bk0mk1_bk0nk1_mn_xdlops_v2r4r2 ...@@ -566,8 +566,8 @@ struct GridwiseGemm_bk0mk1_bk0nk1_mn_xdlops_v2r4r2
template <bool HasMainKBlockLoop, template <bool HasMainKBlockLoop,
InMemoryDataOperationEnum CGlobalMemoryDataOperation, InMemoryDataOperationEnum CGlobalMemoryDataOperation,
typename Block2CTileMap> typename Block2CTileMap>
__device__ static void Run(const FloatAB* p_a_grid, __device__ static void Run(const FloatA* p_a_grid,
const FloatAB* p_b_grid, const FloatB* p_b_grid,
FloatC* p_c_grid, FloatC* p_c_grid,
index_t M, index_t M,
index_t N, index_t N,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment