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,115 @@
+/* ******************************************************************************
+ *
+ *
+ * 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 <helpers/OpTracker.h>
+#include <helpers/logger.h>
+#include <legacy/NativeOps.h>
+
+#include <sstream>
+
+using namespace sd::ops;
+using namespace sd::graph;
+
+namespace sd {
+
+OpTracker& OpTracker::getInstance() {
+  static OpTracker instance;
+  return instance;
+}
+
+void OpTracker::storeOperation(::graph::OpType opType, const OpDescriptor& descriptor) {
+  // check out CPU features
+  if (!isMinimalRequirementsMet()) {
+    auto binaryLevel = ::binaryLevel();
+    auto optimalLevel = ::optimalLevel();
+
+    switch (binaryLevel) {
+      case 3: {
+        sd_printf(
+            "libnd4j binary was built with AVX512 support, but current CPU doesn't have this instruction set. Exiting "
+            "now...",
+            "");
+      } break;
+      case 2: {
+        sd_printf(
+            "libnd4j binary was built with AVX/AVX2 support, but current CPU doesn't have this instruction set. "
+            "Exiting now...",
+            "");
+      } break;
+      default: {
+        sd_printf("Unknown binary validation error. Exiting now...", "");
+      } break;
+    }
+
+    // we're exiting now
+    exit(119);
+  }
+  //
+  if (_map.count(opType) < 1) {
+    std::vector<OpDescriptor> vec;
+    _map[opType] = vec;
+  }
+
+  _operations++;
+
+  auto vec = _map[opType];
+
+  if (std::find(vec.begin(), vec.end(), descriptor) == vec.end()) _map[opType].emplace_back(descriptor);
+}
+
+void OpTracker::storeOperation(::graph::OpType opType, const char* opName, const LongType opNum) {
+  OpDescriptor descriptor(0, opName, false);
+  descriptor.setOpNum((int)opNum);
+  descriptor.setHash(-1);
+
+  storeOperation(opType, descriptor);
+}
+
+template <typename T>
+std::string OpTracker::local_to_string(T value) {
+  std::ostringstream os;
+  os << value;
+  return os.str();
+}
+
+int OpTracker::totalGroups() { return (int)_map.size(); }
+
+int OpTracker::totalOperations() { return _operations; }
+
+const char* OpTracker::exportOperations() {
+  if (_export.length() == 0) {
+    for (auto& v : _map) {
+      std::string block = local_to_string(v.first) + " ";
+
+      for (auto& i : v.second) {
+        block += local_to_string(i.getHash()) + ":";
+        block += local_to_string(i.getOpNum()) + ":";
+        block += *i.getOpName() + "<<";
+      }
+
+      block += ">>";
+      _export += block;
+    }
+  }
+
+  return _export.c_str();
+}
+}  // namespace sd