".github/workflows/cuda-ci.yml" did not exist on "ca1138c32ef521bb97ba5ef4f1fd46779d939829"
Commit 8399b302 authored by Paul's avatar Paul
Browse files

Formatting

parent 7dd2cf04
......@@ -11,10 +11,7 @@ inline namespace MIGRAPHX_INLINE_NS {
namespace gpu {
namespace device {
argument logsoftmax(hipStream_t stream,
argument result,
argument arg,
int axis)
argument logsoftmax(hipStream_t stream, argument result, argument arg, int axis)
{
auto lens = result.get_shape().lens();
......@@ -40,7 +37,7 @@ argument logsoftmax(hipStream_t stream,
for(std::size_t j = 0; j < num_in_batch; ++j)
{
data_idx[axis] = j;
data_idx[axis] = j;
output[data_idx] = input[data_idx] - batch_max;
}
......
......@@ -10,10 +10,7 @@ inline namespace MIGRAPHX_INLINE_NS {
namespace gpu {
namespace device {
argument logsoftmax(hipStream_t stream,
argument result,
argument arg,
int axis);
argument logsoftmax(hipStream_t stream, argument result, argument arg, int axis);
} // namespace device
} // namespace gpu
......
......@@ -15,9 +15,8 @@ shape hip_logsoftmax::compute_shape(const std::vector<shape>& inputs) const
return op.compute_shape({inputs.at(0)});
}
argument hip_logsoftmax::compute(context& ctx,
const shape&,
const std::vector<argument>& args) const
argument
hip_logsoftmax::compute(context& ctx, const shape&, const std::vector<argument>& args) const
{
return device::logsoftmax(ctx.get_stream().get(), args[1], args[0], op.axis);
}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment