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
e711702d
Commit
e711702d
authored
Sep 07, 2021
by
ltqin
Browse files
remove some unused funtion
parent
71467cfc
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
0 additions
and
29 deletions
+0
-29
composable_kernel/include/tensor_operation/gridwise_gemm_xdlops_v2r4.hpp
...el/include/tensor_operation/gridwise_gemm_xdlops_v2r4.hpp
+0
-29
No files found.
composable_kernel/include/tensor_operation/gridwise_gemm_xdlops_v2r4.hpp
View file @
e711702d
...
...
@@ -196,27 +196,6 @@ struct GridwiseGemm_k0mk1_k0nk1_mn_xdlops_v2r4
(
M
%
MPerBlock
==
0
&&
N
%
NPerBlock
==
0
&&
K0
%
KPerBlock
==
0
);
}
__host__
__device__
static
constexpr
index_t
CalculateKBatch
(
const
CMNGridDesc
&
c_m_n_grid_desc
,
const
BK0NK1GridDesc
&
b_k0_n_k1_grid_desc
)
{
constexpr
auto
MAX_GRID
=
2048
;
const
index_t
grid_size_mn
=
CalculateMNGridSize
(
c_m_n_grid_desc
);
const
auto
K0
=
b_k0_n_k1_grid_desc
.
GetLength
(
I0
);
auto
batch
=
K0
/
KPerBlock
;
assert
(
K0
%
KPerBlock
==
0
);
index_t
div
=
1
;
while
(
batch
*
grid_size_mn
>
MAX_GRID
&&
batch
>
div
)
{
div
++
;
if
(
batch
%
div
==
0
)
batch
=
batch
/
div
;
}
batch
=
std
::
max
(
1
,
batch
);
return
batch
;
}
__host__
__device__
static
constexpr
index_t
CalculateGridSize
(
const
CMNGridDesc
&
c_m_n_grid_desc
)
{
...
...
@@ -228,14 +207,6 @@ struct GridwiseGemm_k0mk1_k0nk1_mn_xdlops_v2r4
return
grid_size
;
}
__host__
__device__
static
constexpr
index_t
CalculateBatchGridSize
(
const
index_t
M
,
const
index_t
N
)
{
const
index_t
grid_size_mn
=
(
M
/
MPerBlock
)
*
(
N
/
NPerBlock
);
return
grid_size_mn
;
}
__host__
__device__
static
constexpr
auto
MakeABK0MK1GridDescriptor
(
const
AK0MK1GridDesc
&
a_k0_m_k1_grid_desc
)
{
...
...
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