[Bug] Disable cpu fp16 (#2783)

* disable cpu fp16 * spell mistakes

[Bug] Disable cpu fp16 (#2783)
* disable cpu fp16 * spell mistakes
0b57ce18 · Quan (Andy) Gan · GitHub · 03482f0a · 0b57ce18 · 0b57ce18
Unverified Commit 0b57ce18 authored Mar 25, 2021 by Quan (Andy) Gan Committed by GitHub Mar 25, 2021
Show whitespace changes
Inline Side-by-side

Showing with 7 additions and 5 deletions

src/array/cpu/sddmm.cc src/array/cpu/sddmm.cc +1 -1

src/array/cpu/spmm_binary_ops.h src/array/cpu/spmm_binary_ops.h +4 -2

src/array/cuda/utils.h src/array/cuda/utils.h +2 -2

No files found.
--- 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)

--- 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)

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