Commit 77bdb740 authored by Jing Zhang's avatar Jing Zhang
Browse files

clean

parent 092f54b4
...@@ -123,7 +123,7 @@ __global__ void ...@@ -123,7 +123,7 @@ __global__ void
ignore = barrier_count_finished; ignore = barrier_count_finished;
GridwiseGemm:: GridwiseGemm::
template Run2<HasMainKBlockLoop, GemmSpec, AsLayout, BsLayout, DsLayout, ELayout>( template Run<HasMainKBlockLoop, GemmSpec, AsLayout, BsLayout, DsLayout, ELayout>(
p_as_grid_, p_as_grid_,
p_bs_grid_, p_bs_grid_,
p_ds_grid_, p_ds_grid_,
......
...@@ -956,22 +956,22 @@ struct GridwiseGemmMultipleABD_xdl_cshuffle ...@@ -956,22 +956,22 @@ struct GridwiseGemmMultipleABD_xdl_cshuffle
typename DsLayout, typename DsLayout,
typename ELayout, typename ELayout,
typename Block2ETileMap> typename Block2ETileMap>
__device__ static void Run2(AsGridPointer p_as_grid, __device__ static void Run(AsGridPointer p_as_grid,
BsGridPointer p_bs_grid, BsGridPointer p_bs_grid,
DsGridPointer p_ds_grid, DsGridPointer p_ds_grid,
void* __restrict__ p_e_grid_, void* __restrict__ p_e_grid_,
void* __restrict__ p_shared, void* __restrict__ p_shared,
const AElementwiseOperation& a_element_op, const AElementwiseOperation& a_element_op,
const BElementwiseOperation& b_element_op, const BElementwiseOperation& b_element_op,
const CDEElementwiseOperation& cde_element_op, const CDEElementwiseOperation& cde_element_op,
const index_t M, const index_t M,
const index_t N, const index_t N,
const index_t K, const index_t K,
const std::array<index_t, NumATensor> StrideAs, const std::array<index_t, NumATensor> StrideAs,
const std::array<index_t, NumBTensor> StrideBs, const std::array<index_t, NumBTensor> StrideBs,
const std::array<index_t, NumDTensor> StrideDs, const std::array<index_t, NumDTensor> StrideDs,
const index_t StrideE, const index_t StrideE,
const Block2ETileMap& block_2_etile_map) const Block2ETileMap& block_2_etile_map)
{ {
using AsGridDesc_M_K = using AsGridDesc_M_K =
remove_cvref_t<decltype(MakeAsGridDescriptor_M_K<AsLayout, GemmSpec>({}, {}, {}))>; remove_cvref_t<decltype(MakeAsGridDescriptor_M_K<AsLayout, GemmSpec>({}, {}, {}))>;
......
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