Commit 85cc7cd8 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

fix build error

parent c2aa86c1
......@@ -25,7 +25,7 @@ argument gather(hipStream_t stream,
const auto* inptr = device_cast(input.data());
if(output_shape.scalar())
{
gs_launch(stream, 1)([=](auto i) { outptr[i] = inptr[indices_ptr[0]]; });
gs_launch(stream, 1)([=](auto i) { outptr[i] = inptr[static_cast<int>(indices_ptr[0])]; });
}
else
{
......
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