"tests/git@developer.sourcefind.cn:xdb4_94051/vllm.git" did not exist on "703e42ee4b3efed3c71e7ae7d15f0f96e05722d4"
Commit 0281a411 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

fix review comments

parent 4351f46c
...@@ -19,7 +19,7 @@ argument scatter( ...@@ -19,7 +19,7 @@ argument scatter(
hip_visit_all(result, arg0, inds)([&](auto output, auto data, auto s1) { hip_visit_all(result, arg0, inds)([&](auto output, auto data, auto s1) {
auto* output_ptr = device_cast(output.data()); auto* output_ptr = device_cast(output.data());
const auto* data_ptr = device_cast(data.data()); const auto* data_ptr = device_cast(data.data());
gs_launch(stream, ds.elements(), 256)([=](auto i) gs_launch(stream, ds.elements())([=](auto i)
__device__ { output_ptr[i] = data_ptr[i]; }); __device__ { output_ptr[i] = data_ptr[i]; });
hip_visit_all(arg1, arg2)([&](auto indices, auto update) { 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());
......
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