ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-06 15:54:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move thread per block to the parameter of constructor

Browse Source
This commit is contained in:
rocking
2022-05-18 09:01:44 +08:00
parent 83f7531364
commit c4d610be4f
4 changed files with 21 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
example/19_binary_elementwise/elementwise_add_4d.cpp
View File

@@ -82,8 +82,7 @@ int main()
ck::to_int_vector(a_m.mDesc.GetStrides()),
ck::to_int_vector(b_m.mDesc.GetStrides()),
ck::to_int_vector(c_m.mDesc.GetStrides()),
Add{},
256);
Add{});
if(!broadcastAdd.IsSupportedArgument(argument.get()))
{