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. // Make our own __half definition that is similar to CUDA's.
struct __half { struct __half {
__half() {} EIGEN_DEVICE_FUNC __half() {}
explicit __half(unsigned short raw) : x(raw) {} explicit EIGEN_DEVICE_FUNC __half(unsigned short raw) : x(raw) {}
unsigned short x; unsigned short x;
}; };