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
96af4557
"vscode:/vscode.git/clone" did not exist on "8194199d79c6fbf7f9654c0d25aaaf33f5b0da17"
Commit
96af4557
authored
Jul 18, 2023
by
turneram
Browse files
Add prints to checkvalidity
parent
48717006
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
0 deletions
+6
-0
include/ck/tensor_operation/gpu/grid/gridwise_gemm_multiple_d_xdl_cshuffle.hpp
...ration/gpu/grid/gridwise_gemm_multiple_d_xdl_cshuffle.hpp
+6
-0
No files found.
include/ck/tensor_operation/gpu/grid/gridwise_gemm_multiple_d_xdl_cshuffle.hpp
View file @
96af4557
...
@@ -273,6 +273,7 @@ struct GridwiseGemmMultipleD_xdl_cshuffle
...
@@ -273,6 +273,7 @@ struct GridwiseGemmMultipleD_xdl_cshuffle
// check consistency of desc
// check consistency of desc
if
(
!
(
M
==
e_grid_desc_m_n
.
GetLength
(
I0
)
&&
N
==
e_grid_desc_m_n
.
GetLength
(
I1
)))
if
(
!
(
M
==
e_grid_desc_m_n
.
GetLength
(
I0
)
&&
N
==
e_grid_desc_m_n
.
GetLength
(
I1
)))
{
{
printf
(
"e_grid_desc invalid
\n
"
);
return
false
;
return
false
;
}
}
...
@@ -285,12 +286,14 @@ struct GridwiseGemmMultipleD_xdl_cshuffle
...
@@ -285,12 +286,14 @@ struct GridwiseGemmMultipleD_xdl_cshuffle
if
(
!
valid
)
if
(
!
valid
)
{
{
printf
(
"ds_grid_desc invalid
\n
"
);
return
false
;
return
false
;
}
}
// check tile size
// check tile size
if
(
!
(
M
%
MPerBlock
==
0
&&
N
%
NPerBlock
==
0
&&
K
%
KPerBlock
==
0
))
if
(
!
(
M
%
MPerBlock
==
0
&&
N
%
NPerBlock
==
0
&&
K
%
KPerBlock
==
0
))
{
{
printf
(
"tile size invalid
\n
"
);
return
false
;
return
false
;
}
}
...
@@ -299,12 +302,14 @@ struct GridwiseGemmMultipleD_xdl_cshuffle
...
@@ -299,12 +302,14 @@ struct GridwiseGemmMultipleD_xdl_cshuffle
if
(
!
GridwiseGemmPipe
::
IsSupported
(
num_k_loop
))
if
(
!
GridwiseGemmPipe
::
IsSupported
(
num_k_loop
))
{
{
printf
(
"num_k_loop invalid
\n
"
);
return
false
;
return
false
;
}
}
// check block-to-E-tile
// check block-to-E-tile
if
(
!
block_2_etile_map
.
CheckValidity
(
e_grid_desc_m_n
))
if
(
!
block_2_etile_map
.
CheckValidity
(
e_grid_desc_m_n
))
{
{
printf
(
"block_2_etile_map invalid
\n
"
);
return
false
;
return
false
;
}
}
...
@@ -316,6 +321,7 @@ struct GridwiseGemmMultipleD_xdl_cshuffle
...
@@ -316,6 +321,7 @@ struct GridwiseGemmMultipleD_xdl_cshuffle
b_grid_desc_n_k
.
GetElementSpaceSize
()
*
sizeof
(
ABDataType
)
<=
TwoGB
&&
b_grid_desc_n_k
.
GetElementSpaceSize
()
*
sizeof
(
ABDataType
)
<=
TwoGB
&&
e_grid_desc_m_n
.
GetElementSpaceSize
()
*
sizeof
(
EDataType
)
<=
TwoGB
))
e_grid_desc_m_n
.
GetElementSpaceSize
()
*
sizeof
(
EDataType
)
<=
TwoGB
))
{
{
printf
(
"invalid tensor (> 2GB)
\n
"
);
return
false
;
return
false
;
}
}
...
...
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