"...resnet50_tensorflow.git" did not exist on "577867c6e09417b115ae4e6d0789e9b91d9aa873"
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, ...@@ -25,7 +25,7 @@ argument gather(hipStream_t stream,
const auto* inptr = device_cast(input.data()); const auto* inptr = device_cast(input.data());
if(output_shape.scalar()) 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 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