21 #ifndef EIGEN_INTEROP_HEADERS_SYCL_H 22 #define EIGEN_INTEROP_HEADERS_SYCL_H 26 #if !defined(EIGEN_DONT_VECTORIZE_SYCL) 30 template <
int has_blend,
int lengths>
31 struct sycl_packet_traits : default_packet_traits {
78 #ifdef SYCL_DEVICE_ONLY 79 #define SYCL_PACKET_TRAITS(packet_type, has_blend, unpacket_type, lengths) \ 81 struct packet_traits<unpacket_type> \ 82 : sycl_packet_traits<has_blend, lengths> { \ 83 typedef packet_type type; \ 84 typedef packet_type half; \ 87 SYCL_PACKET_TRAITS(cl::sycl::cl_float4, 1,
float, 4)
88 SYCL_PACKET_TRAITS(cl::sycl::cl_float4, 1, const
float, 4)
89 SYCL_PACKET_TRAITS(cl::sycl::cl_double2, 0,
double, 2)
90 SYCL_PACKET_TRAITS(cl::sycl::cl_double2, 0, const
double, 2)
91 #undef SYCL_PACKET_TRAITS 96 #define SYCL_ARITHMETIC(packet_type) \ 98 struct is_arithmetic<packet_type> { \ 99 enum { value = true }; \ 101 SYCL_ARITHMETIC(cl::sycl::cl_float4)
102 SYCL_ARITHMETIC(cl::sycl::cl_double2)
103 #undef SYCL_ARITHMETIC 105 #define SYCL_UNPACKET_TRAITS(packet_type, unpacket_type, lengths) \ 107 struct unpacket_traits<packet_type> { \ 108 typedef unpacket_type type; \ 109 enum { size = lengths, vectorizable = true, alignment = Aligned16 }; \ 110 typedef packet_type half; \ 112 SYCL_UNPACKET_TRAITS(cl::sycl::cl_float4,
float, 4)
113 SYCL_UNPACKET_TRAITS(cl::sycl::cl_double2,
double, 2)
115 #undef SYCL_UNPACKET_TRAITS 122 namespace TensorSycl {
125 template <
typename PacketReturnType,
int PacketSize>
126 struct PacketWrapper;
128 #ifndef SYCL_DEVICE_ONLY 129 template <
typename PacketReturnType,
int PacketSize>
130 struct PacketWrapper {
131 typedef typename ::Eigen::internal::unpacket_traits<PacketReturnType>::type
133 template <
typename Index>
134 EIGEN_DEVICE_FUNC
static Scalar scalarize(
Index, PacketReturnType &) {
135 eigen_assert(
false &&
"THERE IS NO PACKETIZE VERSION FOR THE CHOSEN TYPE");
138 EIGEN_DEVICE_FUNC
static PacketReturnType convert_to_packet_type(Scalar in,
140 return ::Eigen::internal::template plset<PacketReturnType>(in);
142 EIGEN_DEVICE_FUNC
static void set_packet(PacketReturnType, Scalar *) {
143 eigen_assert(
false &&
"THERE IS NO PACKETIZE VERSION FOR THE CHOSEN TYPE");
148 #elif defined(SYCL_DEVICE_ONLY) 149 template <
typename PacketReturnType>
150 struct PacketWrapper<PacketReturnType, 4> {
151 typedef typename ::Eigen::internal::unpacket_traits<PacketReturnType>::type
153 template <
typename Index>
154 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
static Scalar scalarize(
Index index, PacketReturnType &in) {
167 __builtin_unreachable();
169 __builtin_unreachable();
172 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
static PacketReturnType convert_to_packet_type(
173 Scalar in, Scalar other) {
174 return PacketReturnType(in, other, other, other);
176 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
static void set_packet(PacketReturnType &lhs, Scalar *rhs) {
177 lhs = PacketReturnType(rhs[0], rhs[1], rhs[2], rhs[3]);
181 template <
typename PacketReturnType>
182 struct PacketWrapper<PacketReturnType, 1> {
183 typedef typename ::Eigen::internal::unpacket_traits<PacketReturnType>::type
185 template <
typename Index>
186 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
static Scalar scalarize(
Index, PacketReturnType &in) {
189 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
static PacketReturnType convert_to_packet_type(Scalar in,
191 return PacketReturnType(in);
193 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
static void set_packet(PacketReturnType &lhs, Scalar *rhs) {
198 template <
typename PacketReturnType>
199 struct PacketWrapper<PacketReturnType, 2> {
200 typedef typename ::Eigen::internal::unpacket_traits<PacketReturnType>::type
202 template <
typename Index>
203 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
static Scalar scalarize(
Index index, PacketReturnType &in) {
212 __builtin_unreachable();
214 __builtin_unreachable();
217 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
static PacketReturnType convert_to_packet_type(
218 Scalar in, Scalar other) {
219 return PacketReturnType(in, other);
221 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
static void set_packet(PacketReturnType &lhs, Scalar *rhs) {
222 lhs = PacketReturnType(rhs[0], rhs[1]);
232 #endif // EIGEN_INTEROP_HEADERS_SYCL_H Namespace containing all symbols from the Eigen library.
Definition: Core:141
EIGEN_DEFAULT_DENSE_INDEX_TYPE Index
The Index type as used for the API.
Definition: Meta.h:74
Definition: Eigen_Colamd.h:50