[SYCL] Fix vec::as<vec<bool, N>>() (#9460)

Author: jzc

sycl/include/sycl/detail/generic_type_lists.hpp

@@ -525,6 +525,9 @@ using scalar_bool_list = type_list<bool>;
 
 using bool_list = type_list<scalar_bool_list, marray_bool_list>;
 
+using vector_bool_list = type_list<vec<bool, 1>, vec<bool, 2>, vec<bool, 3>,
+                                   vec<bool, 4>, vec<bool, 8>, vec<bool, 16>>;
+
 // basic types
 using scalar_signed_basic_list =
     type_list<scalar_floating_list, scalar_signed_integer_list>;

sycl/include/sycl/types.hpp

@@ -947,7 +947,8 @@ template <typename Type, int NumElements> class vec {
                   "The new SYCL vec type must have the same storage size in "
                   "bytes as this SYCL vec");
     static_assert(
-        detail::is_contained<asT, detail::gtl::vector_basic_list>::value,
+        detail::is_contained<asT, detail::gtl::vector_basic_list>::value ||
+            detail::is_contained<asT, detail::gtl::vector_bool_list>::value,
         "asT must be SYCL vec of a different element type and "
         "number of elements specified by asT");
     asT Result;
@@ -1959,7 +1960,8 @@ class SwizzleOp {
                   "The new SYCL vec type must have the same storage size in "
                   "bytes as this SYCL swizzled vec");
     static_assert(
-        detail::is_contained<asT, detail::gtl::vector_basic_list>::value,
+        detail::is_contained<asT, detail::gtl::vector_basic_list>::value ||
+            detail::is_contained<asT, detail::gtl::vector_bool_list>::value,
         "asT must be SYCL vec of a different element type and "
         "number of elements specified by asT");
     return Tmp.template as<asT>();

sycl/test/basic_tests/vectors/vectors.cpp

@@ -146,6 +146,10 @@ int main() {
   sycl::vec<int8_t, 2> inputVec = sycl::vec<int8_t, 2>(0, 1);
   auto asVec = inputVec.template swizzle<sycl::elem::s0, sycl::elem::s1>()
                    .template as<sycl::vec<int16_t, 1>>();
+  auto test = inputVec.as<sycl::vec<bool, 2>>();
+  assert(!test[0] && test[1]);
+  assert((inputVec.yx().as<sycl::vec<bool, 2>>()[0]));
+  assert((!inputVec.yx().as<sycl::vec<bool, 2>>()[1]));
 
   // Check that [u]long[n] type aliases match vec<[u]int64_t, n> types.
   assert((std::is_same<sycl::vec<std::int64_t, 2>, sycl::long2>::value));