Skip to content

Commit 1a834b2

Browse files
authored
Fix linalg iterator. (dmlc#8603) (dmlc#8639)
1 parent 162b48a commit 1a834b2

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/common/linalg_op.h

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -62,27 +62,27 @@ void ElementWiseKernel(GenericParameter const* ctx, linalg::TensorView<T, D> t,
6262
#endif // !defined(XGBOOST_USE_CUDA)
6363

6464
template <typename T, std::int32_t kDim>
65-
auto cbegin(TensorView<T, kDim> v) { // NOLINT
65+
auto cbegin(TensorView<T, kDim> const& v) { // NOLINT
6666
auto it = common::MakeIndexTransformIter([&](size_t i) -> std::remove_cv_t<T> const& {
6767
return linalg::detail::Apply(v, linalg::UnravelIndex(i, v.Shape()));
6868
});
6969
return it;
7070
}
7171

7272
template <typename T, std::int32_t kDim>
73-
auto cend(TensorView<T, kDim> v) { // NOLINT
73+
auto cend(TensorView<T, kDim> const& v) { // NOLINT
7474
return cbegin(v) + v.Size();
7575
}
7676

7777
template <typename T, std::int32_t kDim>
78-
auto begin(TensorView<T, kDim> v) { // NOLINT
78+
auto begin(TensorView<T, kDim>& v) { // NOLINT
7979
auto it = common::MakeIndexTransformIter(
8080
[&](size_t i) -> T& { return linalg::detail::Apply(v, linalg::UnravelIndex(i, v.Shape())); });
8181
return it;
8282
}
8383

8484
template <typename T, std::int32_t kDim>
85-
auto end(TensorView<T, kDim> v) { // NOLINT
85+
auto end(TensorView<T, kDim>& v) { // NOLINT
8686
return begin(v) + v.Size();
8787
}
8888
} // namespace linalg

0 commit comments

Comments
 (0)