Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
composable_kernel
Commits
b0e02b8a
Commit
b0e02b8a
authored
May 07, 2023
by
Po-Yen, Chen
Browse files
Hide computing logic of derived attributes
parent
c93c1041
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
8 additions
and
33 deletions
+8
-33
include/ck/tensor_operation/gpu/device/impl/device_cgemm_4gemm_xdl_cshuffle.hpp
...ation/gpu/device/impl/device_cgemm_4gemm_xdl_cshuffle.hpp
+1
-11
include/ck/tensor_operation/gpu/device/impl/device_gemm_xdl_cshuffle.hpp
...or_operation/gpu/device/impl/device_gemm_xdl_cshuffle.hpp
+1
-11
include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdl_cshuffle_v1.hpp
...nsor_operation/gpu/grid/gridwise_gemm_xdl_cshuffle_v1.hpp
+6
-11
No files found.
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
index_t
StrideA_
,
index_t
StrideB_
,
index_t
StrideC_
)
:
Parent
(
M_
,
N_
,
K_
,
StrideA_
,
StrideB_
,
StrideC_
,
GridwiseGemm
::
CalculateMPadded
(
M_
),
GridwiseGemm
::
CalculateNPadded
(
N_
),
GridwiseGemm
::
CalculateKPadded
(
K_
),
GridwiseGemm
::
CalculateAK0
(
K_
),
GridwiseGemm
::
CalculateBK0
(
K_
)),
:
Parent
(
M_
,
N_
,
K_
,
StrideA_
,
StrideB_
,
StrideC_
),
p_a_grid_real
{
p_a_grid_real_
},
p_a_grid_imag
{
p_a_grid_imag_
},
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,
index_t
StrideA_
,
index_t
StrideB_
,
index_t
StrideC_
)
:
Parent
(
M_
,
N_
,
K_
,
StrideA_
,
StrideB_
,
StrideC_
,
GridwiseGemm
::
CalculateMPadded
(
M_
),
GridwiseGemm
::
CalculateNPadded
(
N_
),
GridwiseGemm
::
CalculateKPadded
(
K_
),
GridwiseGemm
::
CalculateAK0
(
K_
),
GridwiseGemm
::
CalculateBK0
(
K_
)),
:
Parent
(
M_
,
N_
,
K_
,
StrideA_
,
StrideB_
,
StrideC_
),
p_a_grid
{
p_a_grid_
},
p_b_grid
{
p_b_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
index_t
K_
,
index_t
StrideA_
,
index_t
StrideB_
,
index_t
StrideC_
,
index_t
MPadded_
,
index_t
NPadded_
,
index_t
KPadded_
,
index_t
AK0_
,
index_t
BK0_
)
index_t
StrideC_
)
:
M
{
M_
},
N
{
N_
},
K
{
K_
},
StrideA
{
StrideA_
},
StrideB
{
StrideB_
},
StrideC
{
StrideC_
},
MPadded
{
MPadded
_
},
NPadded
{
NPadded
_
},
KPadded
{
KPadded
_
},
AK0
{
AK0_
},
BK0
{
BK0_
}
MPadded
{
Calculate
MPadded
(
M_
)
},
NPadded
{
Calculate
NPadded
(
N_
)
},
KPadded
{
Calculate
KPadded
(
K_
)
},
AK0
{
CalculateAK0
(
K_
)
},
BK0
{
CalculateBK0
(
K_
)
}
{
}
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment