"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "59b5985e46c47f97cc13cd175e1d9160061adc8c"
Commit 346366d2 authored by Antoine Kaufmann's avatar Antoine Kaufmann
Browse files

lib/simbricks/base: add SimbricksBaseIfSHMSize()

Determines how much shm a particular connection will need based on the
parameters..
parent f2a67deb
......@@ -153,6 +153,12 @@ void SimbricksBaseIfDefaultParams(struct SimbricksBaseIfParams *params)
params->upper_layer_proto = SIMBRICKS_PROTO_ID_BASE;
}
size_t SimbricksBaseIfSHMSize(struct SimbricksBaseIfParams *params)
{
return params->in_num_entries * params->in_entries_size +
params->out_num_entries * params->out_entries_size;
}
int SimbricksBaseIfInit(struct SimbricksBaseIf *base_if,
struct SimbricksBaseIfParams *params)
{
......
......@@ -124,6 +124,9 @@ int SimbricksBaseIfSHMPoolUnlink(struct SimbricksBaseIfSHMPool *pool);
/** Initialize params struct with default values */
void SimbricksBaseIfDefaultParams(struct SimbricksBaseIfParams *params);
/** Required SHM size for these parameters */
size_t SimbricksBaseIfSHMSize(struct SimbricksBaseIfParams *params);
int SimbricksBaseIfInit(struct SimbricksBaseIf *base_if,
struct SimbricksBaseIfParams *params);
......
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