ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-06 07:51:52 +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/broadcast_add_2d.cpp
View File

@@ -101,8 +101,7 @@ int main()
{Stride, 1},
{0, 1}, // broadcast in first dimension
{Stride, 1},
Add{},
256);
Add{});
if(!broadcastAdd.IsSupportedArgument(argument.get()))
{

3
example/19_binary_elementwise/elementwise_add_1d.cpp
View File

@@ -80,8 +80,7 @@ int main()
{1},
{1},
{1},
Add{},
256);
Add{});
if(!broadcastAdd.IsSupportedArgument(argument.get()))
{

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()))
{