internal static (KVStore, bool) CreateKVStore(string kvstore, int num_device, NDArrayDict arg_params)
{
KVStore kV = null;
var update_on_kvstore = true;
if (num_device == 1 && !kvstore.Contains("dist"))
{
kV = null;
}
else
{
kV = KVStoreBase.Create(kvstore);
if (kvstore == "local")
{
var max_size = arg_params.Values.Select(x => x.Shape.Size).ToList().Max();
if (max_size > 1024 * 1024 * 16)
{
update_on_kvstore = false;
}
}
}
return(kV, update_on_kvstore);
}
internal static (KVStore, bool) CreateSparseKVStore(string kvstore)
{
return(CreateSparseKVStore(KVStoreBase.Create(kvstore)));
}