/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/core/detail/vector_utils.hpp Source File

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/core/detail/vector_utils.hpp Source File#

3 min read time

Applies to Linux

rocCV: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/core/detail/vector_utils.hpp Source File
vector_utils.hpp
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2025 Advanced Micro Devices, Inc. All rights reserved.
3  * Permission is hereby granted, free of charge, to any person obtaining a copy
4  * of this software and associated documentation files (the "Software"), to deal
5  * in the Software without restriction, including without limitation the rights
6  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
7  * copies of the Software, and to permit persons to whom the Software is
8  * furnished to do so, subject to the following conditions:
9  *
10  * The above copyright notice and this permission notice shall be included in
11  * all copies or substantial portions of the Software.
12  *
13  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
14  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
15  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
16  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
17  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
18  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
19  * THE SOFTWARE.
20  */
21 
22 #pragma once
23 
24 #include "core/detail/type_traits.hpp"
25 
26 namespace roccv::detail {
27 
35 template <typename T, typename Base = BaseType<T>, class = std::enable_if_t<HasTypeTraits<T>>>
36 __device__ __host__ constexpr T SetAll(Base v) {
37  if constexpr (!IsCompound<T>) {
38  return v;
39  } else {
40  if constexpr (NumElements<T> == 1)
41  return (T){v};
42  else if constexpr (NumElements<T> == 2)
43  return (T){v, v};
44  else if constexpr (NumElements<T> == 3)
45  return (T){v, v, v};
46  else if constexpr (NumElements<T> == 4)
47  return (T){v, v, v, v};
48  }
49 }
50 } // namespace roccv::detail
roccv::detail
Definition: strided_data_wrap.hpp:33
roccv::detail::SetAll
__device__ constexpr __host__ T SetAll(Base v)
Create a vectorized type with all values set to v.
Definition: vector_utils.hpp:36
type_traits.hpp