minor fix for recent PR (#255)

* minor fix

* clean

[ROCm/composable_kernel commit: 61851ae2b9]
This commit is contained in:
Chao Liu
2022-05-24 21:51:34 -05:00
committed by GitHub
parent 050fc62872
commit 8f61662afc
2 changed files with 3 additions and 4 deletions

View File

@@ -325,7 +325,7 @@ struct DynamicBuffer
{
if(is_valid_element)
{
atomic_add(c_style_pointer_cast<X*>(&p_data_[i]), x);
atomic_add<X>(c_style_pointer_cast<X*>(&p_data_[i]), x);
}
}
}

View File

@@ -26,8 +26,7 @@ int main(int argc, char* argv[])
{
if(strcmp(argv[1], "gemm") == 0)
{
int stat = profile_gemm(argc, argv);
return stat;
return profile_gemm(argc, argv);
}
else if(strcmp(argv[1], "gemm_bias_2d") == 0)
{
@@ -55,7 +54,7 @@ int main(int argc, char* argv[])
}
else if(strcmp(argv[1], "grouped_gemm") == 0)
{
profile_grouped_gemm(argc, argv);
return profile_grouped_gemm(argc, argv);
}
else if(strcmp(argv[1], "conv_fwd") == 0)
{