[SYCL] Implement broadcasting vec::operator=

Signed-off-by: Mariya Podchishchaeva <mariya.podchishchaeva@intel.com>
intel · Apr 4, 2019 · e8ac887 · e8ac887
1 parent 3e11f59
commit e8ac887
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/sycl/include/CL/sycl/types.hpp b/sycl/include/CL/sycl/types.hpp
@@ -372,12 +372,34 @@ template <typename Type, int NumElements> class vec {
     return *this;
   }
 
+#ifdef __SYCL_USE_EXT_VECTOR_TYPE__
+  explicit vec(const DataT &arg) {
+    m_Data = (DataType)arg;
+  }
+
+  template <typename Ty = DataT>
+  typename std::enable_if<std::is_fundamental<Ty>::value, vec &>::type
+  operator=(const DataT &Rhs) {
+    m_Data = (DataType)Rhs;
+    return *this;
+  }
+#else
   explicit vec(const DataT &arg) {
     for (int i = 0; i < NumElements; ++i) {
       setValue(i, arg);
     }
   }
 
+  template <typename Ty = DataT>
+  typename std::enable_if<std::is_fundamental<Ty>::value, vec &>::type
+  operator=(const DataT &Rhs) {
+    for (int i = 0; i < NumElements; ++i) {
+      setValue(i, Rhs);
+    }
+    return *this;
+  }
+#endif
+
 #ifdef __SYCL_USE_EXT_VECTOR_TYPE__
   // Optimized naive constructors with NumElements of DataT values.
   // We don't expect compilers to optimize vararg recursive functions well.

diff --git a/sycl/test/basic_tests/vectors.cpp b/sycl/test/basic_tests/vectors.cpp
@@ -50,5 +50,13 @@ int main() {
   int64_t(vec_2.x());
   cl::sycl::int4(vec_2.x());
 
+  // Check broadcasting operator=
+  cl::sycl::vec<float, 4> b_vec(1.0);
+  b_vec = 0.5;
+  assert(static_cast<float>(b_vec.x()) == static_cast<float>(0.5));
+  assert(static_cast<float>(b_vec.y()) == static_cast<float>(0.5));
+  assert(static_cast<float>(b_vec.z()) == static_cast<float>(0.5));
+  assert(static_cast<float>(b_vec.w()) == static_cast<float>(0.5));
+
   return 0;
 }