FP16 on CUDA are only available starting with cuda 7.5. Disable them when using an older version of CUDA

This commit is contained in:
Benoit Steiner
2016-02-18 23:15:23 -08:00
parent f36c0c2c65
commit 0606a0a39b
3 changed files with 11 additions and 2 deletions

View File

@@ -14,6 +14,8 @@ namespace Eigen {
namespace internal {
#if defined(EIGEN_HAS_CUDA_FP16)
template<>
struct scalar_cast_op<float, half> {
EIGEN_EMPTY_STRUCT_CTOR(scalar_cast_op)
@@ -92,6 +94,7 @@ template<> EIGEN_STRONG_INLINE half2 pcast<float4, half2>(const float4& a) {
#endif
}
#endif
} // end namespace internal