SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
network_builder.h
1 #ifndef _CORE_NETWORK_BUILDER_H_
2 #define _CORE_NETWORK_BUILDER_H_
3 
4 #include <string>
5 
6 #include "smaug/core/workspace.h"
7 #include "smaug/core/network.h"
9 
10 namespace smaug {
11 
24 Network* buildNetwork(const std::string& modelTopoFile,
25  const std::string& modelParamsFile,
26  SamplingInfo& sampling,
27  Workspace* workspace);
28 } // namespace smaug
29 
30 #endif
_SamplingInfo
Simulation sampling information maintained by the Operator and passed to the accelerated kernel.
Definition: common.h:262
smaug::buildNetwork
Network * buildNetwork(const std::string &modelTopoFile, const std::string &modelParamsFile, SamplingInfo &sampling, Workspace *workspace)
buildNetwork reads the specified model topology and parameters protobufs and simulation sampling dire...
Definition: network_builder.cpp:370
smaug
The smaug namespace is the parent namespace of all C++ code in SMAUG.
Definition: backend.cpp:38
common.h
Utilities for writing and invoking Aladdin kernels from Operators.