chore: import upstream snapshot with attribution

2026-07-13 12:47:05 +08:00
commit 4f3b7da785
7394 changed files with 2005594 additions and 0 deletions
@@ -0,0 +1,133 @@
+/* ******************************************************************************
+ *
+ *
+ * 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