wehub-resource-sync
134 lines
4.6 KiB
C++
134 lines
4.6 KiB
C++
/* ******************************************************************************
|
|
*
|
|
*
|
|
* This program and the accompanying materials are made available under the
|
|
* terms of the Apache License, Version 2.0 which is available at
|
|
* https://www.apache.org/licenses/LICENSE-2.0.
|
|
*
|
|
* See the NOTICE file distributed with this work for additional
|
|
* information regarding copyright ownership.
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
* License for the specific language governing permissions and limitations
|
|
* under the License.
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
******************************************************************************/
|
|
|
|
//
|
|
// @author raver119@gmail.com
|
|
//
|
|
#include <graph/profiling/NodeProfile.h>
|
|
#include <helpers/ShapeUtils.h>
|
|
#include <helpers/logger.h>
|
|
|
|
namespace sd {
|
|
namespace graph {
|
|
NodeProfile::NodeProfile(int id, const char *name) {
|
|
_id = id;
|
|
|
|
if (name != nullptr) _name = name;
|
|
};
|
|
|
|
void NodeProfile::printOut() {
|
|
sd_printf("Node: <%i:%s>\n", _id, _name.c_str());
|
|
sd_printf(" Memory: ACT: %lld; TMP: %lld; OBJ: %lld; TTL: %lld;\n", _memoryActivations / _merges,
|
|
_memoryTemporary / _merges, _memoryObjects / _merges, _memoryTotal / _merges);
|
|
sd_printf(" Time: PREP: %lld ns; EXEC: %lld ns; TTL: %lld ns;\n", _preparationTime / _merges,
|
|
_executionTime / _merges, _totalTime / _merges);
|
|
sd_printf(" PREP: INPUT: %lld ns; SHAPE: %lld ns; ARRAY: %lld ns;\n", _inputTime / _merges, _shapeTime / _merges,
|
|
_arrayTime / _merges);
|
|
|
|
std::string inputs;
|
|
std::string outputs;
|
|
|
|
int cnt = 0;
|
|
for (const auto &v : _inputShapes) inputs += v + " ";
|
|
|
|
for (const auto &v : _outputShapes) outputs += v + " ";
|
|
|
|
sd_printf(" Inputs: %s\n", inputs.c_str());
|
|
sd_printf(" Outputs: %s\n", outputs.c_str());
|
|
};
|
|
|
|
LongType NodeProfile::getActivationsSize() const { return _memoryActivations; }
|
|
|
|
void NodeProfile::setShapeFunctionTime(LongType time) { _shapeTime = time; }
|
|
|
|
void NodeProfile::setArrayTime(LongType time) { _arrayTime = time; }
|
|
|
|
void NodeProfile::setInputTime(LongType time) { _inputTime = time; }
|
|
|
|
LongType NodeProfile::getTemporarySize() const { return _memoryTemporary; }
|
|
|
|
LongType NodeProfile::getObjectsSize() const { return _memoryObjects; }
|
|
|
|
LongType NodeProfile::getTotalSize() const { return _memoryTotal; }
|
|
|
|
void NodeProfile::setBuildTime(LongType time) { _buildTime = time; }
|
|
|
|
void NodeProfile::setPreparationTime(LongType time) { _preparationTime = time; }
|
|
|
|
void NodeProfile::setExecutionTime(LongType time) { _executionTime = time; }
|
|
|
|
void NodeProfile::setTotalTime(LongType time) { _totalTime = time; }
|
|
|
|
void NodeProfile::setActivationsSize(LongType bytes) { _memoryActivations = bytes; }
|
|
|
|
void NodeProfile::setTemporarySize(LongType bytes) { _memoryTemporary = bytes; }
|
|
|
|
void NodeProfile::setObjectsSize(LongType bytes) { _memoryObjects = bytes; }
|
|
|
|
void NodeProfile::setTotalSize(LongType bytes) { _memoryTotal = bytes; }
|
|
|
|
LongType NodeProfile::getExecutionTime() const { return _executionTime; }
|
|
|
|
void NodeProfile::addInputShape(LongType const *shapeInfo) {
|
|
_inputShapes.emplace_back(ShapeUtils::shapeInfoAsString(shapeInfo));
|
|
}
|
|
|
|
void NodeProfile::addOutputShape(LongType const *shapeInfo) {
|
|
_outputShapes.emplace_back(ShapeUtils::shapeInfoAsString(shapeInfo));
|
|
}
|
|
|
|
void NodeProfile::merge(NodeProfile *other) {
|
|
_merges += other->_merges;
|
|
_memoryObjects += other->_memoryObjects;
|
|
_memoryActivations += other->_memoryActivations;
|
|
_memoryTemporary += other->_memoryTemporary;
|
|
_memoryTotal += other->_memoryTotal;
|
|
|
|
_preparationTime += other->_preparationTime;
|
|
_executionTime += other->_executionTime;
|
|
_totalTime += other->_totalTime;
|
|
_shapeTime += other->_shapeTime;
|
|
_arrayTime += other->_arrayTime;
|
|
_inputTime += other->_inputTime;
|
|
|
|
_inputShapes = other->_inputShapes;
|
|
_outputShapes = other->_outputShapes;
|
|
}
|
|
|
|
std::string &NodeProfile::name() { return _name; }
|
|
|
|
void NodeProfile::assign(NodeProfile *other) {
|
|
_merges = other->_merges;
|
|
_memoryObjects = other->_memoryObjects;
|
|
_memoryActivations = other->_memoryActivations;
|
|
_memoryTemporary = other->_memoryTemporary;
|
|
_memoryTotal = other->_memoryTotal;
|
|
|
|
_preparationTime = other->_preparationTime;
|
|
_executionTime = other->_executionTime;
|
|
_totalTime = other->_totalTime;
|
|
_shapeTime = other->_shapeTime;
|
|
_arrayTime = other->_arrayTime;
|
|
_inputTime = other->_inputTime;
|
|
|
|
_inputShapes = other->_inputShapes;
|
|
_outputShapes = other->_outputShapes;
|
|
}
|
|
} // namespace graph
|
|
} // namespace sd
|