Added missing EIGEN_DEVICE_FUNC

This commit is contained in:
Benoit Steiner
2016-05-11 14:10:09 -07:00

View File

@@ -46,8 +46,8 @@
// Make our own __half definition that is similar to CUDA's.
struct __half {
__half() {}
explicit __half(unsigned short raw) : x(raw) {}
EIGEN_DEVICE_FUNC __half() {}
explicit EIGEN_DEVICE_FUNC __half(unsigned short raw) : x(raw) {}
unsigned short x;
};