Skip to content

GitLab

"tests/python/common/data/test_data.py" did not exist on "dc78e11c3f590536b000cc86dad9ae6173539767"
Unverified Commit 0b57ce18 authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

[Bug] Disable cpu fp16 (#2783) 

* disable cpu fp16

* spell mistakes
parent 03482f0a
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 5 deletions
src/array/cpu/sddmm.cc
View file @ 0b57ce18
...@@ -50,7 +50,7 @@ namespace aten { ...@@ -50,7 +50,7 @@ namespace aten {
typedef double DType; \ typedef double DType; \
{ __VA_ARGS__ } \ { __VA_ARGS__ } \
} else { \ } else { \
LOG(FATAL) << "Data type not renogized with bits " << bits; \ LOG(FATAL) << "Data type not recognized with bits " << bits; \
} \ } \
} while (0) } while (0)
......
src/array/cpu/spmm_binary_ops.h
View file @ 0b57ce18
...@@ -148,14 +148,16 @@ constexpr DType Min<DType>::zero; ...@@ -148,14 +148,16 @@ constexpr DType Min<DType>::zero;
#define SWITCH_BITS(bits, DType, ...) \ #define SWITCH_BITS(bits, DType, ...) \
do { \ do { \
if ((bits) == 16 || (bits) == 32) { \ if ((bits) == 16) { \
LOG(FATAL) << "FP16 not supported on CPU"; \
} else if ((bits) == 32) { \
typedef float DType; \ typedef float DType; \
{ __VA_ARGS__ } \ { __VA_ARGS__ } \
} else if ((bits) == 64) { \ } else if ((bits) == 64) { \
typedef double DType; \ typedef double DType; \
{ __VA_ARGS__ } \ { __VA_ARGS__ } \
} else { \ } else { \
LOG(FATAL) << "Data type not renogized with bits " << bits; \ LOG(FATAL) << "Data type not recognized with bits " << bits; \
} \ } \
} while (0) } while (0)
......
src/array/cuda/utils.h
View file @ 0b57ce18
...@@ -31,7 +31,7 @@ namespace cuda { ...@@ -31,7 +31,7 @@ namespace cuda {
typedef double DType; \ typedef double DType; \
{ __VA_ARGS__ } \ { __VA_ARGS__ } \
} else { \ } else { \
LOG(FATAL) << "Data type not renogized with bits " << bits; \ LOG(FATAL) << "Data type not recognized with bits " << bits; \
} \ } \
} while (0) } while (0)
#else // USE_FP16 #else // USE_FP16
...@@ -44,7 +44,7 @@ namespace cuda { ...@@ -44,7 +44,7 @@ namespace cuda {
typedef double DType; \ typedef double DType; \
{ __VA_ARGS__ } \ { __VA_ARGS__ } \
} else { \ } else { \
LOG(FATAL) << "Data type not renogized with bits " << bits; \ LOG(FATAL) << "Data type not recognized with bits " << bits; \
} \ } \
} while (0) } while (0)
#endif // USE_FP16 #endif // USE_FP16
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment