Skip to content

GitLab

"torchvision/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "75e4a7dd384776f5722a35588020f73eb8dc0575"
Commit b0e02b8a authored by Po-Yen, Chen's avatar Po-Yen, Chen
Browse files

Hide computing logic of derived attributes

parent c93c1041
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 33 deletions
include/ck/tensor_operation/gpu/device/impl/device_cgemm_4gemm_xdl_cshuffle.hpp
View file @ b0e02b8a
...@@ -185,17 +185,7 @@ struct DeviceCGemm_4Gemm_Xdl_CShuffle ...@@ -185,17 +185,7 @@ struct DeviceCGemm_4Gemm_Xdl_CShuffle
index_t StrideA_, index_t StrideA_,
index_t StrideB_, index_t StrideB_,
index_t StrideC_) index_t StrideC_)
: Parent(M_, : Parent(M_, N_, K_, StrideA_, StrideB_, StrideC_),
N_,
K_,
StrideA_,
StrideB_,
StrideC_,
GridwiseGemm::CalculateMPadded(M_),
GridwiseGemm::CalculateNPadded(N_),
GridwiseGemm::CalculateKPadded(K_),
GridwiseGemm::CalculateAK0(K_),
GridwiseGemm::CalculateBK0(K_)),
p_a_grid_real{p_a_grid_real_}, p_a_grid_real{p_a_grid_real_},
p_a_grid_imag{p_a_grid_imag_}, p_a_grid_imag{p_a_grid_imag_},
p_b_grid_real{p_b_grid_real_}, p_b_grid_real{p_b_grid_real_},
......
include/ck/tensor_operation/gpu/device/impl/device_gemm_xdl_cshuffle.hpp
View file @ b0e02b8a
...@@ -143,17 +143,7 @@ struct DeviceGemm_Xdl_CShuffle : public DeviceGemm<ALayout, ...@@ -143,17 +143,7 @@ struct DeviceGemm_Xdl_CShuffle : public DeviceGemm<ALayout,
index_t StrideA_, index_t StrideA_,
index_t StrideB_, index_t StrideB_,
index_t StrideC_) index_t StrideC_)
: Parent(M_, : Parent(M_, N_, K_, StrideA_, StrideB_, StrideC_),
N_,
K_,
StrideA_,
StrideB_,
StrideC_,
GridwiseGemm::CalculateMPadded(M_),
GridwiseGemm::CalculateNPadded(N_),
GridwiseGemm::CalculateKPadded(K_),
GridwiseGemm::CalculateAK0(K_),
GridwiseGemm::CalculateBK0(K_)),
p_a_grid{p_a_grid_}, p_a_grid{p_a_grid_},
p_b_grid{p_b_grid_}, p_b_grid{p_b_grid_},
p_c_grid{p_c_grid_} p_c_grid{p_c_grid_}
......
include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdl_cshuffle_v1.hpp
View file @ b0e02b8a
...@@ -395,23 +395,18 @@ struct GridwiseGemm_k0mk1_k0nk1_mn_xdl_cshuffle_v1 ...@@ -395,23 +395,18 @@ struct GridwiseGemm_k0mk1_k0nk1_mn_xdl_cshuffle_v1
index_t K_, index_t K_,
index_t StrideA_, index_t StrideA_,
index_t StrideB_, index_t StrideB_,
index_t StrideC_, index_t StrideC_)
index_t MPadded_,
index_t NPadded_,
index_t KPadded_,
index_t AK0_,
index_t BK0_)
: M{M_}, : M{M_},
N{N_}, N{N_},
K{K_}, K{K_},
StrideA{StrideA_}, StrideA{StrideA_},
StrideB{StrideB_}, StrideB{StrideB_},
StrideC{StrideC_}, StrideC{StrideC_},
MPadded{MPadded_}, MPadded{CalculateMPadded(M_)},
NPadded{NPadded_}, NPadded{CalculateNPadded(N_)},
KPadded{KPadded_}, KPadded{CalculateKPadded(K_)},
AK0{AK0_}, AK0{CalculateAK0(K_)},
BK0{BK0_} BK0{CalculateBK0(K_)}
{ {
} }
......
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