MCPcopy Create free account

hub / github.com/Avarok-Cybersecurity/atlas / functions

Functions7,049 in github.com/Avarok-Cybersecurity/atlas

↓ 8 callersMethodinto_parser
Create a boxed parser for this format.
crates/spark-server/src/reasoning_parser/mod.rs:101
↓ 8 callersMethodis_enabled
Marconi prefix-cache region availability.
crates/spark-model/src/model/ssm_snapshot.rs:192
↓ 8 callersFunctionkernel_modules_for_dtype
Module + function name 4-tuple consumed by `Qwen3AttentionLayer::new_with_gating`: `(reshape_mod, reshape_fn, decode_mod, decode_fn)`. The reshape pai
crates/spark-model/src/layers/qwen3_attention/init_kernel_dispatch.rs:28
↓ 8 callersMethodlayer_prefix
Weight key prefix for layer-level weights. Returns `"model.layers"` for flat models (qwen3_next), or `"model.language_model.layers"` for conditional g
crates/atlas-core/src/config/methods.rs:154
↓ 8 callersFunctionliteral
(bytes: &[u8])
crates/xgrammar/src/fsm/algorithms_tests.rs:11
↓ 8 callersFunctionmask
A mask with `n` accepted indices — `memory_size()` is `4*n` bytes, so tests can dial the cache's memory accounting precisely.
crates/xgrammar/src/compiler/rule_cache_tests.rs:17
↓ 8 callersFunctionms
(x: u64)
crates/spark-server/src/scheduler/mtp_gate.rs:311
↓ 8 callersMethodname
(&self)
crates/spark-server/src/scheduler/logit_processors/mid_word.rs:44
↓ 8 callersMethodnum_attention_layers
Number of full attention layers.
crates/atlas-core/src/config/methods.rs:37
↓ 8 callersMethodpredequant_for_prefill
Pre-dequant dense (non-expert) NVFP4 weights to FP8 for zero-overhead prefill. Only affects gate GEMM and shared expert GEMMs. Expert weights stay N
crates/spark-model/src/layers/moe/helpers_c.rs:12
↓ 8 callersFunctionrope
RoPE: apply rotary position embeddings to Q and K in-place. Kernel: `rope_forward(Q, K, positions, seq_len, num_q_heads, num_kv_heads, head_dim, rota
crates/spark-model/src/layers/ops/embeddings.rs:24
↓ 8 callersFunctionsample_with_params
Convenience wrapper: sample without token history (no repetition penalty).
crates/spark-runtime/src/sampler.rs:459
↓ 8 callersFunctionsection
(label: str)
tests/test_openai_compat_v2.py:64
↓ 8 callersFunctionsend_streaming
(url: &str, prompt: &str, max_tokens: usize)
crates/atlas-spark-bench/src/lib.rs:155
↓ 8 callersMethodset_start_state
Set the start state.
crates/xgrammar/src/fsm/with_start_end.rs:72
↓ 8 callersFunctionstructural_tag_to_grammar
Convert a structural-tag JSON document into a grammar. Port of `StructuralTagToGrammar` / `Grammar::FromStructuralTag(json)`. The document must be a
crates/xgrammar/src/structural_tag/mod.rs:80
↓ 8 callersMethodsync_secondary
Wait for all work on the secondary stream to complete.
crates/spark-model/src/traits/model.rs:627
↓ 8 callersFunctiontest_tool_defs
()
crates/spark-server/src/grammar/tests/mod.rs:37
↓ 8 callersMethodtoken_ids
Token IDs `[M]` u32 — stable across the layer loop (DeepSeek-V4 hash-MoE reads `tid2eid[token_id]`). Upload the pass's token IDs here before the layer
crates/spark-runtime/src/buffers.rs:275
↓ 8 callersMethodtruncate_after
Drop every entry whose `token_position` is strictly greater than `keep_len`. Called after a successful rollback: snapshots taken in the now-discarded
crates/spark-server/src/scheduler/ssm_decode_ring.rs:149
↓ 8 callersFunctionurl_of
(a: &Annotation)
crates/spark-server/src/openai/tests.rs:5
↓ 8 callersFunctionvision_dense_auto
( store: &WeightStore, prefix: &str, gpu: &dyn GpuBackend, )
crates/spark-model/src/weight_loader/qwen35.rs:19
↓ 8 callersFunctionwith_local
( _f: impl FnOnce(&mut HighSpeedSwap) -> anyhow::Result<R>, )
crates/spark-storage/src/stubs.rs:96
↓ 8 callersMethodwrite_from_host
(&mut self, key: GroupKey, src: &[u8])
crates/spark-storage/src/backend/posix.rs:57
↓ 7 callersMethodadd_rule_with_allocated_name
Add a rule under a name allocated via [`Self::allocate_rule_name`]. Returns the rule name. Port of `AddRuleWithAllocatedName`.
crates/xgrammar/src/schema/script.rs:53
↓ 7 callersMethodadvance_bytes
Advance past `n` *bytes* of escape input. Because every escape sequence `parse_next_escaped` consumes is pure ASCII, one byte equals one codepoint her
crates/xgrammar/src/regex/cursor.rs:93
↓ 7 callersFunctionalloc_or_managed
(gpu: &dyn GpuBackend, bytes: usize)
crates/spark-model/src/weight_loader/deepseek_v4/compute.rs:9
↓ 7 callersFunctionanthropic_error
(status: StatusCode, error_type: &str, message: String)
crates/spark-server/src/anthropic/helpers.rs:10
↓ 7 callersFunctionavg
(key)
bench/quick_bench.py:197
↓ 7 callersMethodbegin_step
Reset (m, l, o) for `num_seqs` sequences. Call once at the start of each decode step before the first `step_tile`.
crates/spark-storage/src/tiled_attention.rs:123
↓ 7 callersFunctionbf16_to_f32
(lo: u8, hi: u8)
crates/spark-server/src/scheduler/helpers.rs:7
↓ 7 callersFunctionbuild_leaf_fsm
Build a leaf FSM from a regex fragment that is either a plain literal (`"abx"`, possibly with escapes / `.`) or a character class `[...]`. `regex` is
crates/xgrammar/src/fsm/builder/regex_leaf.rs:45
↓ 7 callersFunctionbuild_projection
Random Gaussian projection matrix `P` of shape `[head_dim, r]`, BF16, in row-major layout. Variance 1/head_dim so that ⟨k, p⟩ has unit variance for un
crates/spark-storage/src/projection.rs:32
↓ 7 callersFunctionbuild_trie
Build a trie-based FSM. `patterns` are inserted as a character trie. `excluded_patterns` are only honored when `add_back_edges` is true (the trie bec
crates/xgrammar/src/fsm/builder/trie.rs:29
↓ 7 callersFunctionbuilt
(ebnf: &str)
crates/xgrammar/src/grammar/functor/fsm_builder_tests.rs:6
↓ 7 callersFunctionchat
(messages, max_tokens=50, temperature=0.0)
scripts/bench_model.py:12
↓ 7 callersMethodclaim_guarded
Claim a slot and wrap it in a [`SlotGuard`] that returns the slot to the free list when dropped. This is the leak-safe claim API: the guard is stored
crates/spark-model/src/model/ssm_pool.rs:168
↓ 7 callersMethodcompile_minimax_xml_tool_grammar
F66 (2026-04-29): MiniMax M2.7 XML tool-call grammar. Native MiniMax format: ```xml <minimax:tool_call> <invoke name="tool_name"> <parameter name="ke
crates/spark-server/src/grammar/compile_tools.rs:693
↓ 7 callersMethodcompile_structural_tag
Compile a structural tag. Port of `GrammarCompiler::CompileStructuralTag`. Delegates the structural-tag-JSON -> `GrammarData` conversion to the W5 `s
crates/xgrammar/src/compiler/compiler.rs:228
↓ 7 callersMethodconvert
Run the conversion — port of `RegexConverter::Convert`.
crates/xgrammar/src/regex/converter.rs:53
↓ 7 callersFunctionconvert_python_jinja_to_minijinja
Convert Python Jinja2 syntax to minijinja-compatible syntax. Handles: slice reversal, string methods, etc.
crates/spark-server/src/tokenizer/jinja_helpers.rs:373
↓ 7 callersFunctioncopy_d2h
(dst: &mut [T], src: *const c_void)
crates/spark-runtime/src/cutlass/tests/mod.rs:58
↓ 7 callersMethoddecomposition
The WGRAMMAR static/dynamic decomposition (Tier 3c) — the compile-time split of every rule body into fixed scaffolding spans (with precomputed literal
crates/xgrammar/src/compiler/compiled_grammar.rs:220
↓ 7 callersMethoddelete
Remove an entry. Returns `true` when the id existed and matched `kind`, `false` otherwise. Also removes the entry from the persistent backend when one
crates/spark-server/src/response_store/store_impl.rs:167
↓ 7 callersMethoddtype
(&self)
crates/spark-runtime/src/kv_cache/paged_impl.rs:436
↓ 7 callersMethodends
The accepting-state bitmap.
crates/xgrammar/src/fsm/with_start_end.rs:57
↓ 7 callersFunctionenforce_min_length_on_required_strings
( schema: &serde_json::Value, )
crates/spark-server/src/grammar/schema.rs:60
↓ 7 callersMethodfill_next_token_bitmask
Fill the next-token bitmask into `bitmask` (`index` selects the matcher's slice when several share one buffer; pass `0` for a single matcher). Returns
crates/xgrammar/src/api/matcher.rs:95
↓ 7 callersMethodfinalize
Divide o_state by l_state and store as BF16 in `output`.
crates/spark-storage/src/tiled_attention.rs:274
↓ 7 callersMethodforward_batched
Batched forward: GEMM gate for N tokens, per-token expert dispatch. Gate projection reads weights once for N tokens (GEMM M=N). Expert dispatch remai
crates/spark-model/src/layers/moe/forward_batched.rs:12
↓ 7 callersMethodforward_prefill
( &self, input: DevicePtr, // [num_tokens, H] BF16 — normed MoE input num_tokens: usiz
crates/spark-model/src/layers/moe/forward_prefill.rs:17
↓ 7 callersFunctionfp32_from_bits
kernels/gb10/qwen3.6-27b/nvfp4/q4k_vendor/ggml-impl.h:366
↓ 7 callersFunctionfp8_gemm_t_blockscaled
( gpu: &dyn GpuBackend, kernel: KernelHandle, a_fp8: DevicePtr, a_scale: DevicePtr, b_fp8:
crates/spark-model/src/layers/ops/gemm_quant.rs:349
↓ 7 callersFunctiongen
(url, prompt)
bench/prefix_cache_contamination.py:58
↓ 7 callersMethodget_by_id
Finish building and return the [`GrammarData`], setting the root rule to `root_rule_id`. Returns `Err` if `root_rule_id` is out of bounds.
crates/xgrammar/src/grammar/builder.rs:126
↓ 7 callersMethodget_rule_id
Rule id of the rule named `name`, or `-1` if absent.
crates/xgrammar/src/grammar/builder_rules.rs:163
↓ 7 callersFunctionhc_head
( gpu: &dyn GpuBackend, kernel: KernelHandle, streams: DevicePtr, head_fn: DevicePtr, head
crates/spark-model/src/layers/ops/hyper_connection.rs:108
↓ 7 callersMethodidx
(&self)
crates/spark-model/src/model/ssm_pool.rs:429
↓ 7 callersFunctionintset_union
Replace `lhs` with the sorted union of `lhs` and `rhs`. Both inputs must already be sorted ascending; duplicates within an input are collapsed in the
crates/xgrammar/src/support/int_set.rs:13
↓ 7 callersMethodis_empty
True if the cache holds no entries.
crates/xgrammar/src/compiler/rule_cache.rs:230
↓ 7 callersMethodis_epsilon
True if this is an epsilon transition.
crates/xgrammar/src/fsm/edge.rs:59
↓ 7 callersMethodis_set
(&self, token: usize)
crates/xgrammar/src/matcher/bitmask.rs:113
↓ 7 callersFunctionjpath
Read 'a.b.c' from nested dict. Returns None on missing.
bench/fp8_dgx2_drift/harness/aggregate.py:57
↓ 7 callersMethodk_block_stride_bytes_for_layer
K-side block stride in bytes for a specific attention layer. Same as `block_stride_bytes_for_layer`; named for clarity in asym call sites.
crates/spark-runtime/src/kv_cache/paged_impl.rs:377
↓ 7 callersMethodlatest_scanable_states
The scanable states reached at the current input position.
crates/xgrammar/src/earley/parser.rs:153
↓ 7 callersMethodlen
Number of signatures currently stored.
crates/spark-server/src/loop_simhash.rs:108
↓ 7 callersFunctionmake_correction
(name: &str, rows: usize, cols: usize, rank: usize)
crates/spark-server/src/lqer.rs:395
↓ 7 callersFunctionmake_pattern_for_digit_range
A digit-range pattern for one fixed-length slot: either a literal digit or a `[a-b]` class, optionally followed by `\d{n}`.
crates/xgrammar/src/schema/range_regex.rs:13
↓ 7 callersMethodmax_batch_tokens
(&self)
crates/spark-runtime/src/buffers.rs:349
↓ 7 callersMethodmaybe_remeasure
Re-open measurement when the live depth has left the regime the current decision was measured in (factor-[`REMEASURE_DEPTH_FACTOR`] crossing in either
crates/spark-server/src/scheduler/mtp_gate.rs:178
↓ 7 callersFunctionmoe_topk_sigmoid_batched
( gpu: &dyn GpuBackend, kernel: KernelHandle, gate_logits: DevicePtr, bias: DevicePtr, exp
crates/spark-model/src/layers/ops/moe_grouped_a.rs:81
↓ 7 callersFunctionmoe_topk_softmax_batched
( gpu: &dyn GpuBackend, kernel: KernelHandle, gate_logits: DevicePtr, expert_indices: DevicePt
crates/spark-model/src/layers/ops/moe_grouped_a.rs:51
↓ 7 callersFunctionmoe_w4a16_grouped_gemm_ptrtable
( gpu: &dyn GpuBackend, kernel: KernelHandle, a: DevicePtr, b_packed_ptrs: DevicePtr, b_sc
crates/spark-model/src/layers/ops/moe_grouped_a.rs:147
↓ 7 callersFunctionnot_supported
Generic 501 "not supported" response used by the auto-probe stubs below. OpenAI-SDK auto-detection and observability wrappers expect a 501 + `error.ty
crates/spark-server/src/api/completions.rs:478
↓ 7 callersMethodor
Join `items` with ` | `, wrapped in parentheses. Port of static `Or`.
crates/xgrammar/src/schema/script.rs:98
↓ 7 callersFunctionparse_one_call
Auto-detect and parse inner content of a `<tool_call>` block. Tries Gemma-4 native, JSON (hermes), qwen3_coder XML, then tag-style XML fallback.
crates/spark-server/src/tool_parser/parse_single_a.rs:8
↓ 7 callersFunctionparse_utf8
Decode every codepoint in a UTF-8 byte slice. When `preserve_invalid_bytes` is `false` (the C++ default), an invalid byte aborts decoding and the res
crates/xgrammar/src/support/encoding.rs:136
↓ 7 callersMethodpeek
Token at `cur + delta`. The token stream is always `EndOfFile`- terminated, so callers stay in bounds when reading near the end.
crates/xgrammar/src/grammar/parser.rs:72
↓ 7 callersMethodpop
Pop `count` advanced states (rollback).
crates/xgrammar/src/compiler/mask_gen/mod.rs:138
↓ 7 callersMethodprefill
(&self, tokens: &[u32], seq: &mut SequenceState, stream: u64)
crates/spark-model/src/model/trait_impl/mod.rs:55
↓ 7 callersFunctionprimary_arg_for_tool
(name: &str, args_json: &str)
crates/spark-server/src/api/sanitizer.rs:333
↓ 7 callersFunctionprint_as_escaped
Render a single codepoint as a printable, escaped string. `additional_escape_map` is checked first (e.g. `{'-': "\\-"}`), then the C-escape table, th
crates/xgrammar/src/support/escape.rs:68
↓ 7 callersMethodpush_in_latest_row
Append one element to the most recently inserted row. # Panics Panics if there is no row to append to.
crates/xgrammar/src/support/compact_2d_array.rs:124
↓ 7 callersMethodr
(&mut self, lo: f64, hi: f64)
crates/spark-model/examples/gdn_verify_fused_microtest.rs:62
↓ 7 callersMethodrecord_event
(&self, event: u64, stream: u64)
crates/spark-model/src/model/trait_impl/mod.rs:408
↓ 7 callersMethodrepeat
Apply an EBNF repetition suffix to `item`. Port of static `Repeat`. `max == -1` means unbounded.
crates/xgrammar/src/schema/script.rs:117
↓ 7 callersMethodrepeat_edge_info
Decode the repeat-edge info stored at aux index `idx`.
crates/xgrammar/src/fsm/fsm.rs:80
↓ 7 callersFunctionresolve_model_dir
Resolve a model specifier to a local directory path. Resolution order: 1. If the specifier is an existing directory with `config.json`, use it direct
crates/spark-server/src/model_resolver.rs:15
↓ 7 callersFunctionsample_first_token
Sample the FIRST generated token after prefill, applying the grammar constraint when one is active. `#131`: the very first decode token is produced f
crates/spark-server/src/scheduler/sample_step.rs:476
↓ 7 callersFunctionscalar_f32
Extract a scalar f32 from a single-element weight tensor via D2H copy.
crates/spark-model/src/weight_map/model_a.rs:31
↓ 7 callersFunctionsh
(cmd, check=True, capture=False, timeout=None)
tests/ab_35b.py:70
↓ 7 callersFunctionshard_quantized_nvfp4
Shard an NVFP4-quantized weight. The packed weight is `[N, K/2]` u8; the per-group scale is `[N, K/group_size]` u8 (FP8); `weight_scale_2` is a per-te
crates/spark-model/src/tp_shard/quant_shard.rs:36
↓ 7 callersFunctionskip
(name: str, reason: str = "")
scripts/test_coherence.py:58
↓ 7 callersMethodssm_ba_size
SSM beta+alpha projection output size.
crates/atlas-core/src/config/methods.rs:98
↓ 7 callersMethodstep_tile
( &self, ctx: &CudaCtx, q: u64, k_pool: u64, v_pool: u64, tile
crates/spark-storage/src/tiled_attention.rs:181
↓ 7 callersMethodsystem_prompt
(&self, _tools: &[ToolDefinition], _tool_choice: &ToolChoice)
crates/spark-server/src/tool_parser/gemma4.rs:30
↓ 7 callersMethodunion
Parallel union of several FSMs — accepts any of their languages.
crates/xgrammar/src/fsm/fsm_ops.rs:51
↓ 7 callersMethodupdate_lookahead_assertion
Set the lookahead-assertion expr id of rule `rule_id`. An id of `-1` means no assertion.
crates/xgrammar/src/grammar/builder_rules.rs:98
↓ 7 callersFunctionupload
(gpu: &dyn GpuBackend, bytes: &[u8])
crates/spark-model/examples/w4a16_bf16_microtest.rs:104
↓ 7 callersFunctionupload
(gpu: &dyn GpuBackend, bytes: &[u8])
crates/spark-model/examples/w4a16_bf16_v2_microtest.rs:104
← previousnext →501–600 of 7,049, ranked by callers