minor fix for recent PR (#255)

* minor fix * clean

minor fix for recent PR (#255)
* minor fix * clean
61851ae2 · Chao Liu · GitHub · 40b59a63 · 61851ae2 · 61851ae2
Unverified Commit 61851ae2 authored May 24, 2022 by Chao Liu Committed by GitHub May 24, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 4 deletions

include/ck/utility/dynamic_buffer.hpp include/ck/utility/dynamic_buffer.hpp +1 -1

profiler/src/profiler.cpp profiler/src/profiler.cpp +2 -3

No files found.
--- a/include/ck/utility/dynamic_buffer.hpp
+++ b/include/ck/utility/dynamic_buffer.hpp
@@ -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);
            }
        }
    }

--- a/profiler/src/profiler.cpp
+++ b/profiler/src/profiler.cpp
@@ -26,8 +26,7 @@ int main(int argc, char* argv[])
 {
    if(strcmp(argv[1], "gemm") == 0)
    {
-        int stat = profile_gemm(argc, argv);
+        return profile_gemm(argc, argv);
-        return stat;
    }
    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)
    {