Add make_const_array

f8e5a547 · Paul · e2120984 · f8e5a547
Commit f8e5a547 authored Oct 10, 2022 by Paul
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

src/targets/gpu/kernels/include/migraphx/kernels/array.hpp src/targets/gpu/kernels/include/migraphx/kernels/array.hpp +6 -0

No files found.
--- a/src/targets/gpu/kernels/include/migraphx/kernels/array.hpp
+++ b/src/targets/gpu/kernels/include/migraphx/kernels/array.hpp
@@ -272,6 +272,12 @@ struct integral_const_array : array<T, sizeof...(Xs)>
    MIGRAPHX_DEVICE_CONSTEXPR integral_const_array() : base_array({Xs...}) {}
 };

+template <class T, class... Ts>
+constexpr auto make_const_array(T x, Ts... xs)
+{
+    return integral_const_array<typename T::value_type, x, xs...>{};
+}
+
 template <class T, T... Xs, class F>
 constexpr auto transform(integral_const_array<T, Xs...>, F f)
 {