diff --git a/graphbolt/src/cnumpy.cc b/graphbolt/src/cnumpy.cc index 0b6fd8c00232..ccbc45e71792 100644 --- a/graphbolt/src/cnumpy.cc +++ b/graphbolt/src/cnumpy.cc @@ -232,7 +232,7 @@ void OnDiskNpyArray::IndexSelectIOUringImpl( throw std::runtime_error("IndexError: Index out of range."); } - return result; + // return result; the input result parameter is the return value of this func. } c10::intrusive_ptr> OnDiskNpyArray::IndexSelectIOUring( @@ -246,8 +246,8 @@ c10::intrusive_ptr> OnDiskNpyArray::IndexSelectIOUring( .layout(torch::kStrided) .requires_grad(false)); - auto future = - at::intraop_launch_future([=]() { IndexSelectIOUring(index, result); }); + auto future = at::intraop_launch_future( + [=]() { IndexSelectIOUringImpl(index, result); }); return c10::make_intrusive>(future, result); } diff --git a/graphbolt/src/cnumpy.h b/graphbolt/src/cnumpy.h index 81c9ca6443ef..4022d8a58e98 100644 --- a/graphbolt/src/cnumpy.h +++ b/graphbolt/src/cnumpy.h @@ -100,8 +100,7 @@ class OnDiskNpyArray : public torch::CustomClassHolder { c10::intrusive_ptr> IndexSelectIOUring( torch::Tensor index); - void OnDiskNpyArray::IndexSelectIOUringImpl( - torch::Tensor index, torch::Tensor result); + void IndexSelectIOUringImpl(torch::Tensor index, torch::Tensor result); #endif // HAVE_LIBRARY_LIBURING private: