Commit 19690f5b authored by rocking's avatar rocking
Browse files

check gemm vaildity in IsSupportedArgument

parent a3d2a407
...@@ -942,7 +942,11 @@ struct DeviceGemmMultipleDLayernorm_Xdl_CShuffle ...@@ -942,7 +942,11 @@ struct DeviceGemmMultipleDLayernorm_Xdl_CShuffle
} }
} }
return true; return GridwiseGemmWelford::CheckValidity(arg.a_grid_desc_m_k_,
arg.b_grid_desc_n_k_,
arg.ds_grid_desc_m_n_,
arg.gemm_e_grid_desc_m_n_,
arg.block_2_etile_map_);
} }
// polymorphic // polymorphic
......
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