Skip to content

GitLab

Commit fd9e80c5 authored by aska-0096's avatar aska-0096
Browse files

fix gemm

parent 8053bca3
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 7 deletions
include/ck/tensor_operation/gpu/grid/gridwise_gemm_multiple_d_wmma_cshuffle.hpp
View file @ fd9e80c5
...@@ -349,10 +349,10 @@ struct GridwiseGemmMultipleD_Wmma ...@@ -349,10 +349,10 @@ struct GridwiseGemmMultipleD_Wmma
using GridwiseGemmPipe = using GridwiseGemmPipe =
remove_cvref_t<decltype(GridwiseGemmPipeline_Selector<PipelineVer, remove_cvref_t<decltype(GridwiseGemmPipeline_Selector<PipelineVer,
AEnableLds,
BEnableLds,
NumGemmKPrefetchStage, NumGemmKPrefetchStage,
LoopSched>())>; LoopSched,
AEnableLds,
BEnableLds>())>;
// Describe how data store to (LDS/VGPR) buffer from Global memory // Describe how data store to (LDS/VGPR) buffer from Global memory
__host__ __device__ static constexpr auto MakeABlockDescriptor() __host__ __device__ static constexpr auto MakeABlockDescriptor()
......
include/ck/tensor_operation/gpu/grid/gridwise_gemm_wmma.hpp
View file @ fd9e80c5
...@@ -141,11 +141,11 @@ struct GridwiseGemm_Wmma ...@@ -141,11 +141,11 @@ struct GridwiseGemm_Wmma
using ThisThreadBlock = ThisThreadBlock<BlockSize>; using ThisThreadBlock = ThisThreadBlock<BlockSize>;
using GridwiseGemmPipe = using GridwiseGemmPipe =
remove_cvref_t<decltype(GridwiseGemmPipeline_Selector<PipelineVer, remove_cvref_t<decltype(GridwiseGemmPipeline_Selector<PipelineVer,,
AEnableLds,
BEnableLds,
NumGemmKPrefetchStage, NumGemmKPrefetchStage,
LoopSched>())>; LoopSched,
AEnableLds,
BEnableLds>())>;
// Describe how data store to (LDS/VGPR) buffer from Global memory // Describe how data store to (LDS/VGPR) buffer from Global memory
__host__ __device__ static constexpr auto MakeABlockDescriptor() __host__ __device__ static constexpr auto MakeABlockDescriptor()
......
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