Refactor: share HBG Graph Execution across architectures

src/a2a3/runtime/host_build_graph/build_config.py

@@ -24,7 +24,10 @@
 
 BUILD_CONFIG = {
     "aicore": {"include_dirs": ["runtime", "common", ".."], "source_dirs": ["aicore", "orchestration"]},
-    "aicpu": {"include_dirs": ["runtime", "common", ".."], "source_dirs": ["aicpu", "runtime", "orchestration"]},
+    "aicpu": {
+        "include_dirs": ["runtime", "common", ".."],
+        "source_dirs": ["aicpu", "runtime", "orchestration", "../../../common/host_build_graph"],
+    },
     "host": {
         "include_dirs": ["runtime", "common", ".."],
         "source_dirs": ["host", "runtime/orchestrator_core", "runtime/shared", "orchestration"],

src/a2a3/runtime/host_build_graph/runtime/graph_cache.h

@@ -11,76 +11,4 @@
 
 #pragma once
 
-#include <stddef.h>
-#include <stdint.h>
-
-#include <type_traits>
-
-#include "pto_task_id.h"
-#include "pto_types.h"
-
-inline constexpr uint32_t GRAPH_MAX_TENSOR_ARGS = 32;
-
-struct GraphScopeResult {
-    bool execute_block{true};
-    bool recording{false};
-    PTO2TaskId task_id{PTO2TaskId::invalid()};
-};
-
-using GraphSubmitResult = GraphScopeResult;
-
-constexpr uint64_t graph_hash_byte(uint64_t h, uint8_t b) { return (h ^ static_cast<uint64_t>(b)) * 1099511628211ULL; }
-
-inline uint64_t graph_hash_bytes(uint64_t h, const void *data, size_t bytes) {
-    const auto *p = static_cast<const uint8_t *>(data);
-    for (size_t i = 0; i < bytes; ++i) {
-        h = graph_hash_byte(h, p[i]);
-    }
-    return h;
-}
-
-constexpr uint64_t graph_const_hash_impl(const char *s, uint64_t h) {
-    return (*s == '\0') ? h : graph_const_hash_impl(s + 1, graph_hash_byte(h, static_cast<uint8_t>(*s)));
-}
-
-constexpr uint64_t GRAPH_KEY(const char *s) { return graph_const_hash_impl(s, 1469598103934665603ULL); }
-
-inline bool rt_graph_args_cacheable(const CoreTaskArgs &args) {
-    if (args.has_error || args.tensor_count() <= 0 ||
-        args.tensor_count() > static_cast<int32_t>(GRAPH_MAX_TENSOR_ARGS)) {
-        return false;
-    }
-    for (int32_t i = 0; i < args.tensor_count(); ++i) {
-        // A Graph boundary is caller-owned storage. Runtime-allocated
-        // TensorCreateInfo outputs remain on the ordinary submit path.
-        if (args.tag(i) == TensorArgType::OUTPUT) return false;
-    }
-    return true;
-}
-
-inline uint64_t rt_graph_make_key(uint64_t graph_id) { return graph_id; }
-
-template <typename T>
-inline uint64_t graph_hash_config_value(uint64_t hash, T value) {
-    using Value = std::remove_cv_t<std::remove_reference_t<T>>;
-    static_assert(
-        std::is_integral_v<Value> || std::is_same_v<Value, float> || std::is_same_v<Value, double>,
-        "Graph construction parameters must be integral, float, or double values"
-    );
-    constexpr uint8_t category = std::is_same_v<Value, bool> ? 1 :
-                                 std::is_integral_v<Value>   ? (std::is_signed_v<Value> ? 2 : 3) :
-                                                               4;
-    constexpr uint8_t width = sizeof(Value);
-    hash = graph_hash_byte(hash, category);
-    hash = graph_hash_byte(hash, width);
-    return graph_hash_bytes(hash, &value, sizeof(value));
-}
-
-template <typename... Config>
-inline uint64_t rt_graph_make_key(uint64_t graph_id, Config... config) {
-    uint64_t hash = graph_hash_bytes(1469598103934665603ULL, &graph_id, sizeof(graph_id));
-    const uint32_t count = sizeof...(Config);
-    hash = graph_hash_bytes(hash, &count, sizeof(count));
-    ((hash = graph_hash_config_value(hash, config)), ...);
-    return hash;
-}
+#include "host_build_graph/graph_cache.h"