Unverified Commit 1abe87f5 authored by Chao Ma's avatar Chao Ma Committed by GitHub
Browse files

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