devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Improved support for vectorization of 16-bit floats

Browse Source
This commit is contained in:
Benoit Steiner
2016-06-09 08:22:27 -07:00
parent 15890c304e
commit aa33446dac
3 changed files with 95 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
unsupported/Eigen/CXX11/src/Tensor/TensorMeta.h
View File

@@ -54,6 +54,11 @@ struct PacketType {
// For CUDA packet types when using a GpuDevice
#if defined(EIGEN_USE_GPU) && defined(__CUDACC__)
template <>
struct PacketType<half, GpuDevice> {
typedef half2 type;
static const int size = 2;
};
template <>
struct PacketType<float, GpuDevice> {
typedef float4 type;