From 26a932dcde766996442a7d352ba80d7ebe387d5f Mon Sep 17 00:00:00 2001 From: Cliff Burdick <30670611+cliffburdick@users.noreply.github.com> Date: Mon, 24 Feb 2025 13:22:37 -0800 Subject: [PATCH] Revert broadcasting changes to concat --- include/matx/operators/concat.h | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/include/matx/operators/concat.h b/include/matx/operators/concat.h index f7429bc6..040abe16 100644 --- a/include/matx/operators/concat.h +++ b/include/matx/operators/concat.h @@ -91,16 +91,12 @@ namespace matx } } - - - template __MATX_INLINE__ __MATX_DEVICE__ __MATX_HOST__ auto GetVal(cuda::std::array &indices) const { if constexpr ( I == N ) { // This should never happen return value_type{}; - // returning this to satisfy lvalue requirements } else { const auto &op = cuda::std::get(ops_); auto idx = indices[axis_]; @@ -108,7 +104,7 @@ namespace matx // If in range of this operator if(idx < size) { // evaluate operator - return get_value(cuda::std::forward(op), indices); + return cuda::std::apply(op, indices); } else { // otherwise remove this operator and recurse indices[axis_] -= size; @@ -116,7 +112,8 @@ namespace matx } } } - + + template __MATX_INLINE__ __MATX_DEVICE__ __MATX_HOST__ decltype(auto) GetVal(cuda::std::array &indices) { @@ -124,7 +121,7 @@ namespace matx // This should never happen // returning this to satisfy lvalue requirements auto &op = cuda::std::get(ops_); - return get_value(cuda::std::forward(op), indices); + return cuda::std::apply(op, indices); } else { auto &op = cuda::std::get(ops_); auto idx = indices[axis_]; @@ -132,7 +129,7 @@ namespace matx // If in range of this operator if(idx < size) { // evaluate operator - return get_value(cuda::std::forward(op), indices); + return cuda::std::apply(op, indices); } else { // otherwise remove this operator and recurse indices[axis_] -= size;