"vscode:/vscode.git/clone" did not exist on "09157a7275970625b677b1759bc380d90024d34b"
Commit 7135da72 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent 973cafd4
......@@ -10,23 +10,22 @@ inline namespace MIGRAPHX_INLINE_NS {
namespace gpu {
namespace device {
argument scatter(hipStream_t stream, argument result, argument arg0, argument arg1, argument arg2, int64_t axis)
argument scatter(
hipStream_t stream, argument result, argument arg0, argument arg1, argument arg2, int64_t axis)
{
auto ds = arg0.get_shape();
auto ds = arg0.get_shape();
auto inds = arg1.get_shape();
hip_visit_all(result, arg0, inds)([&](auto output, auto data, auto s1) {
// hip_visit_all(result, arg0, arg2, ds)([&](auto output, auto data, auto update, auto s) {
auto* output_ptr = device_cast(output.data());
auto* output_ptr = device_cast(output.data());
const auto* data_ptr = device_cast(data.data());
gs_launch(stream, ds.elements(), 256)([=](auto i) __device__ {
output_ptr[i] = data_ptr[i];
});
gs_launch(stream, ds.elements(), 256)([=](auto i)
__device__ { output_ptr[i] = data_ptr[i]; });
hip_visit_all(arg1, arg2)([&](auto indices, auto update) {
const auto* upd_ptr = device_cast(update.data());
const auto* upd_ptr = device_cast(update.data());
const auto* indices_ptr = device_cast(indices.data());
gs_launch(stream, inds.elements(), 256)([=](auto i) __device__ {
auto out_idx = s1.multi(i);
out_idx[axis] = indices_ptr[i];
auto out_idx = s1.multi(i);
out_idx[axis] = indices_ptr[i];
output[out_idx] = upd_ptr[i];
});
});
......
......@@ -10,7 +10,8 @@ inline namespace MIGRAPHX_INLINE_NS {
namespace gpu {
namespace device {
argument scatter(hipStream_t stream, argument result, argument arg0, argument arg1, argument arg2, int64_t axis);
argument scatter(
hipStream_t stream, argument result, argument arg0, argument arg1, argument arg2, int64_t 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