[KVstore] Change sizeof(float) to dtype.bits / 8 (#1452)

* update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * fix lint * update * update

[KVstore] Change sizeof(float) to dtype.bits / 8 (#1452)
* update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * update * fix lint * update * update
1abe87f5 · Chao Ma · GitHub · 338f24cf · 1abe87f5
Unverified Commit 1abe87f5 authored Apr 17, 2020 by Chao Ma Committed by GitHub Apr 17, 2020
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

src/graph/network.cc src/graph/network.cc +3 -1

No files found.
--- a/src/graph/network.cc
+++ b/src/graph/network.cc
@@ -669,7 +669,7 @@ DGL_REGISTER_GLOBAL("network._CAPI_FastPull")
        row_size *= local_data->shape[i];
      }
    }
-    row_size *= sizeof(float);
+    row_size *= (local_data->dtype.bits / 8);
    // Get local id and remote id
    if (str_flag.compare("has_g2l") == 0) {
      NDArray g2l = args[11];
@@ -716,6 +716,8 @@ DGL_REGISTER_GLOBAL("network._CAPI_FastPull")
 #ifndef _WIN32  // windows does not support rand_r()
        int s_id = (rand_r(&seed) % (higher-lower+1))+lower;
        send_kv_message(sender, &kv_msg, s_id, !AUTO_FREE);
+#else
+        LOG(FATAL) << "KVStore does not support Windows yet.";
 #endif
        msg_count++;
      }