code cleanup

1a51797e · Shucai Xiao · b6786993 · 1a51797e
Commit 1a51797e authored Jun 25, 2019 by Shucai Xiao
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

src/targets/gpu/device/softmax.cpp src/targets/gpu/device/softmax.cpp +2 -2

No files found.
--- a/src/targets/gpu/device/softmax.cpp
+++ b/src/targets/gpu/device/softmax.cpp
@@ -61,7 +61,7 @@ void softmax(hipStream_t stream, const argument& result, const argument& arg, in

                    auto item_num =
                        (remaining_item_num > block_size) ? block_size : remaining_item_num;
-                    reduce_max<type>(lds_data, block_size, thr_idx, item_num);
+                    reduce_max(lds_data, block_size, thr_idx, item_num);

                    remaining_item_num -= block_size;
                }
@@ -84,7 +84,7 @@ void softmax(hipStream_t stream, const argument& result, const argument& arg, in

                    auto item_num =
                        (remaining_item_num > block_size) ? block_size : remaining_item_num;
-                    reduce_sum<type>(lds_data, block_size, thr_idx, item_num);
+                    reduce_sum(lds_data, block_size, thr_idx, item_num);

                    remaining_item_num -= block_size;
                }