Commit d5a32cd2 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

merge changes from device refactor.

parents 89dfc4dd 8be483c5
......@@ -10,7 +10,7 @@ inline namespace MIGRAPHX_INLINE_NS {
namespace gpu {
namespace device {
void logsoftmax(hipStream_t stream, const argument& result, const argument& arg, int axis);
void logsoftmax(hipStream_t stream, argument result, argument arg, int axis);
} // namespace device
} // namespace gpu
......
......@@ -10,7 +10,7 @@ inline namespace MIGRAPHX_INLINE_NS {
namespace gpu {
namespace device {
void softmax(hipStream_t stream, const argument& result, const argument& arg, int axis);
void softmax(hipStream_t stream, argument result, argument arg, int axis);
} // namespace device
} // namespace gpu
......
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