[Bug] Disable cpu fp16 (dmlc#2783)

* disable cpu fp16 * spell mistakes
skorani · Mar 25, 2021 · 0b57ce1 · 0b57ce1
1 parent 03482f0
commit 0b57ce1
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/src/array/cpu/sddmm.cc b/src/array/cpu/sddmm.cc
@@ -50,7 +50,7 @@ namespace aten {
       typedef double DType;                                     \
       { __VA_ARGS__ }                                           \
     } else {                                                    \
-      LOG(FATAL) << "Data type not renogized with bits " << bits; \
+      LOG(FATAL) << "Data type not recognized with bits " << bits; \
     }                                                           \
   } while (0)
 

diff --git a/src/array/cpu/spmm_binary_ops.h b/src/array/cpu/spmm_binary_ops.h
@@ -148,14 +148,16 @@ constexpr DType Min<DType>::zero;
 
 #define SWITCH_BITS(bits, DType, ...)                           \
   do {                                                          \
-    if ((bits) == 16 || (bits) == 32) {                         \
+    if ((bits) == 16) {                                         \
+      LOG(FATAL) << "FP16 not supported on CPU";                \
+    } else if ((bits) == 32) {                                  \
       typedef float DType;                                      \
       { __VA_ARGS__ }                                           \
     } else if ((bits) == 64) {                                  \
       typedef double DType;                                     \
       { __VA_ARGS__ }                                           \
     } else {                                                    \
-      LOG(FATAL) << "Data type not renogized with bits " << bits; \
+      LOG(FATAL) << "Data type not recognized with bits " << bits; \
     }                                                           \
   } while (0)
 

diff --git a/src/array/cuda/utils.h b/src/array/cuda/utils.h
@@ -31,7 +31,7 @@ namespace cuda {
       typedef double DType;                                     \
       { __VA_ARGS__ }                                           \
     } else {                                                    \
-      LOG(FATAL) << "Data type not renogized with bits " << bits; \
+      LOG(FATAL) << "Data type not recognized with bits " << bits; \
     }                                                           \
   } while (0)
 #else  // USE_FP16
@@ -44,7 +44,7 @@ namespace cuda {
       typedef double DType;                                     \
       { __VA_ARGS__ }                                           \
     } else {                                                    \
-      LOG(FATAL) << "Data type not renogized with bits " << bits; \
+      LOG(FATAL) << "Data type not recognized with bits " << bits; \
     }                                                           \
   } while (0)
 #endif  // USE_FP16