Add explicit conversion to vec

celerity · Jan 27, 2025 · bf7fe2a · bf7fe2a
1 parent cc80743
commit bf7fe2a
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/include/simsycl/sycl/vec.hh b/include/simsycl/sycl/vec.hh
@@ -69,6 +69,11 @@ template<typename T, typename DataT, int NumElements>
 concept VecCompatible
     = vec_like_num_elements<DataT, T>::value == 1 || vec_like_num_elements<DataT, T>::value == NumElements;
 
+template<typename From, typename To>
+concept implicitly_convertible = requires { std::is_convertible_v<From, To>; };
+
+template<typename From, typename To>
+concept explicitly_convertible = requires { static_cast<To>(std::declval<From>()); };
 
 template<int... Is>
 struct no_repeat_indices;
@@ -261,6 +266,13 @@ class swizzled_vec {
         return m_elems[indices[0]];
     }
 
+    template<typename T>
+    explicit operator T() const
+        requires(num_elements == 1 && detail::explicitly_convertible<value_type, T>)
+    {
+        return m_elems[indices[0]];
+    }
+
     static constexpr size_t byte_size() noexcept { return sycl::vec<value_type, num_elements>::byte_size(); }
 
     static constexpr size_t size() noexcept { return num_elements; }
@@ -548,6 +560,13 @@ class alignas(detail::vec_alignment_v<DataT, NumElements>) vec {
         return m_elems[0];
     }
 
+    template<typename T>
+    explicit operator T() const
+        requires(NumElements == 1 && detail::explicitly_convertible<DataT, T>)
+    {
+        return m_elems[0];
+    }
+
     static constexpr size_t byte_size() noexcept { return sizeof m_elems; }
 
     static constexpr size_t size() noexcept { return NumElements; }