#include "graph_ir.h"
#include <algorithm>
#include <charconv>
#include <cmath>
#include <cstdlib>
#include <cstring>
#include <fstream>
#include <limits>
#include <optional>
#include <set>
#include <sstream>
#include <stdexcept>
#include <tuple>
#include <type_traits>
#include <variant>
#include "cpu_attention.h"
#include "determinism.h"
#include "gguf_structs.h"
#include "graph_ir_expr.h"
#include "graph_ir_mamba.h"
#include "graph_ir_ops.h"
#include "graph_ir_recurrence.h"
#include "graph_ir_validate.h"
#include "ir_digest.h"
#include "kv_quant.h"
#include "logger.h"
#include "quantization.h"
#include "utils.h"
#include "weight_management.h"
Go to the source code of this file.
|
| void | ember_ir::linear_attention_row_cpu (const float *q, const float *k, const float *v, float *num, float *den, float *out, int width, float den_floor) |
| |
| void | ember_ir::ssm_scan_row_cpu (const float *u, const float *A, const float *B, const float *C, float D, float *s, float *scratch, float *y, int n) |
| |
| bool | ember_ir::has_recurrent_state (const std::string &ir_json) |
| |
| bool | ember_ir::has_windowed_attention (const std::string &ir_json) |
| |
| bool | ember_ir::has_state_tokens (const std::string &ir_json) |
| |
| bool | ember_ir::tuned_llama_match (const std::string &ir_json, const GGUFData &gguf, std::string *why) |
| |
| bool | ember_ir::device_supported (const std::string &ir_json) |
| |
| bool | ember_ir::has_ir_builder (const std::string &arch) |
| |
| std::string | ember_ir::find_ir (const GGUFData &gguf) |
| |
| std::string | ember_ir::build_ir_from_gguf (const GGUFData &g, bool log) |
| |
| static void | ember_ir::cpu_only (bool gpu, const char *what) |
| |
◆ cfg_
◆ d_
◆ doc_
◆ H_
◆ hd_
◆ KV_
◆ role_
Definition at line 433 of file graph_ir.cpp.
433{{"tokens", "tokens"}, {"pos", "pos"}};
◆ slot_
◆ why
◆ x_