clang format

42b24bd1 · Shucai Xiao · b9575730 · 42b24bd1 · 42b24bd1
Commit 42b24bd1 authored Jun 25, 2019 by Shucai Xiao
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 5 deletions

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

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

No files found.
--- a/src/targets/gpu/device/logsoftmax.cpp
+++ b/src/targets/gpu/device/logsoftmax.cpp
@@ -83,8 +83,8 @@ void logsoftmax(hipStream_t stream, argument result, argument arg, int axis)
            for(size_t i = thr_idx; i < batch_item_num; i += block_size)
            {
-                data_idx[axis] = i;
+                data_idx[axis]   = i;
-                output[data_idx]  = input[data_idx] - log_batch_sum;
+                output[data_idx] = input[data_idx] - log_batch_sum;
            }
        });
    });

--- a/src/targets/gpu/device/softmax.cpp
+++ b/src/targets/gpu/device/softmax.cpp
@@ -83,9 +83,9 @@ void softmax(hipStream_t stream, argument result, argument arg, int axis)
            for(size_t i = thr_idx; i < batch_item_num; i += block_size)
            {
-                data_idx[axis] = i;
+                data_idx[axis]   = i;
-                auto val       = input[data_idx] - batch_max;
+                auto val         = input[data_idx] - batch_max;
-                output[data_idx]  = ::exp(to_hip_type(val)) / batch_sum;
+                output[data_idx] = ::exp(to_hip_type(val)) / batch_sum;
            }
        });
    });