Fabio
c36510c21e
Some checks failed
Linux Build / Linux cpu (push) Has been cancelled
Linux Build / Linux cpu-clang (push) Has been cancelled
Linux Build / Linux vulkan (push) Has been cancelled
mac-build / macOS CPU (push) Has been cancelled
Nix Build / Nix (cpu) (push) Has been cancelled
Nix Build / Nix (vulkan) (push) Has been cancelled
windows-build / Windows CPU (push) Has been cancelled
120 lines
3.9 KiB
C++
120 lines
3.9 KiB
C++
#pragma once
|
|
|
|
#include "engine/framework/io/json.h"
|
|
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
#include <initializer_list>
|
|
#include <optional>
|
|
#include <string>
|
|
#include <string_view>
|
|
#include <unordered_map>
|
|
|
|
namespace engine::assets {
|
|
enum class TensorStorageType;
|
|
}
|
|
|
|
namespace engine::core {
|
|
enum class BackendType;
|
|
}
|
|
|
|
namespace engine::runtime {
|
|
|
|
struct OptionMatch {
|
|
std::string key;
|
|
std::string value;
|
|
};
|
|
|
|
struct ComponentExecPlan {
|
|
std::string impl;
|
|
std::optional<core::BackendType> backend;
|
|
std::unordered_map<std::string, std::string> artifacts;
|
|
io::json::Value::Object params;
|
|
};
|
|
|
|
struct ComponentExecImplementation {
|
|
std::string_view name;
|
|
std::initializer_list<core::BackendType> backends;
|
|
std::initializer_list<std::string_view> required_artifacts;
|
|
std::initializer_list<std::string_view> optional_artifacts;
|
|
std::initializer_list<std::string_view> params;
|
|
};
|
|
|
|
struct ComponentExecSpec {
|
|
std::string_view name;
|
|
std::string_view default_impl;
|
|
std::initializer_list<ComponentExecImplementation> implementations;
|
|
};
|
|
|
|
// Parses one family.exec_plan JSON object. Omitted components/fields retain their
|
|
// defaults. Parameter values retain JSON types for model-local semantic validation.
|
|
std::unordered_map<std::string, ComponentExecPlan> parse_exec_plan(
|
|
const std::unordered_map<std::string, std::string> & options,
|
|
std::string_view key,
|
|
std::initializer_list<ComponentExecSpec> components);
|
|
|
|
std::optional<OptionMatch> find_option_match(
|
|
const std::unordered_map<std::string, std::string> & options,
|
|
std::initializer_list<std::string_view> keys);
|
|
|
|
std::optional<std::string> find_option(
|
|
const std::unordered_map<std::string, std::string> & options,
|
|
std::initializer_list<std::string_view> keys);
|
|
|
|
bool parse_bool_option(const std::string & value, std::string_view key);
|
|
|
|
std::optional<int> parse_int_option(
|
|
const std::unordered_map<std::string, std::string> & options,
|
|
std::initializer_list<std::string_view> keys);
|
|
|
|
std::optional<float> parse_float_option(
|
|
const std::unordered_map<std::string, std::string> & options,
|
|
std::initializer_list<std::string_view> keys);
|
|
|
|
std::optional<float> parse_finite_float_option(
|
|
const std::unordered_map<std::string, std::string> & options,
|
|
std::initializer_list<std::string_view> keys);
|
|
|
|
std::optional<float> parse_positive_finite_float_option(
|
|
const std::unordered_map<std::string, std::string> & options,
|
|
std::initializer_list<std::string_view> keys);
|
|
|
|
std::optional<uint32_t> parse_u32_option(
|
|
const std::unordered_map<std::string, std::string> & options,
|
|
std::initializer_list<std::string_view> keys);
|
|
|
|
std::optional<uint64_t> parse_u64_option(
|
|
const std::unordered_map<std::string, std::string> & options,
|
|
std::initializer_list<std::string_view> keys);
|
|
|
|
std::optional<int64_t> parse_i64_option(
|
|
const std::unordered_map<std::string, std::string> & options,
|
|
std::initializer_list<std::string_view> keys);
|
|
|
|
int64_t parse_positive_i64_option(
|
|
const std::unordered_map<std::string, std::string> & options,
|
|
std::initializer_list<std::string_view> keys,
|
|
int64_t fallback);
|
|
|
|
size_t parse_size_mb_option(
|
|
const std::unordered_map<std::string, std::string> & options,
|
|
std::initializer_list<std::string_view> keys,
|
|
size_t fallback_bytes);
|
|
|
|
assets::TensorStorageType parse_tensor_storage_option(
|
|
const std::unordered_map<std::string, std::string> & options,
|
|
std::string_view key,
|
|
assets::TensorStorageType fallback,
|
|
std::initializer_list<assets::TensorStorageType> allowed);
|
|
|
|
assets::TensorStorageType parse_tensor_storage_option(
|
|
const std::unordered_map<std::string, std::string> & options,
|
|
std::string_view key,
|
|
std::string_view fallback_key,
|
|
assets::TensorStorageType fallback,
|
|
std::initializer_list<assets::TensorStorageType> allowed);
|
|
|
|
uint32_t random_u32_seed();
|
|
uint64_t random_u64_seed();
|
|
|
|
} // namespace engine::runtime
|