"vscode:/vscode.git/clone" did not exist on "8194199d79c6fbf7f9654c0d25aaaf33f5b0da17"
Commit 96af4557 authored by turneram's avatar turneram
Browse files

Add prints to checkvalidity

parent 48717006
...@@ -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;
} }
......
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