Added missing EIGEN_DEVICE_FUNC

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

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;
};