MCPcopy Create free account

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

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

⨍ Functions 7,049◇ Types & classes 847
↓ 2,370 callersMethodarg_ptr
Add a DevicePtr (u64) argument.
crates/spark-runtime/src/kernel_args.rs:96
↓ 1,829 callersMethodarg_u32
Add a u32 argument.
crates/spark-runtime/src/kernel_args.rs:106
↓ 952 callersMethoditer
Iterate over rows as slices.
crates/xgrammar/src/support/compact_2d_array.rs:75
↓ 942 callersMethodget
(&self, layer: &str)
crates/spark-server/src/lqer.rs:102
↓ 621 callersMethodoffset
Byte offset from this pointer.
crates/spark-runtime/src/gpu.rs:47
↓ 496 callersMethodkernel
(&self, _module: &str, _func_name: &str)
crates/spark-runtime/src/gpu/mock.rs:152
↓ 483 callersMethodlaunch
(&self, stream: u64)
crates/spark-storage/src/cuda_graph.rs:58
↓ 480 callersMethodblock
(mut self, block: [u32; 3])
crates/spark-runtime/src/kernel_args.rs:85
↓ 480 callersMethodgrid
(mut self, grid: [u32; 3])
crates/spark-runtime/src/kernel_args.rs:80
↓ 464 callersMethodexpect
(self, expected: WeightQuantFormat, context: &str)
crates/spark-model/src/weight_map/quantized.rs:58
↓ 457 callersMethodlen
(&self)
crates/spark-server/src/lqer.rs:106
↓ 393 callersMethodlen
(&self)
crates/spark-server/src/scheduler/ssm_decode_ring.rs:168
↓ 390 callersMethodis_empty
(&self)
crates/spark-server/src/lqer.rs:110
↓ 358 callersMethodalloc
Allocate a cache in the given storage format.
crates/spark-model/src/forward/qwen3_5/mod.rs:309
↓ 279 callersMethodsynchronize
Block the host until all work submitted to `stream` has completed. Used by `mixed_forward_batch` to retire the decode pass (which runs on the default
crates/spark-model/src/traits/model.rs:735
↓ 274 callersFunctiondiv_ceil
(a: u32, b: u32)
crates/spark-model/examples/gemv_fp4_vs_fp8_microtest.rs:33
↓ 267 callersMethodget
Fetch a clone of the stored body + metadata. TTL-expired entries are evicted (and removed from the backend) and treated as missing. Kind mismatch retu
crates/spark-server/src/response_store/store_impl.rs:139
↓ 242 callersMethodcopy_h2d
(&self, src: &[u8], dst: DevicePtr)
crates/spark-runtime/src/gpu/mock.rs:105
↓ 240 callersMethodcontains
Check if a weight exists.
crates/spark-runtime/src/weights.rs:116
↓ 239 callersMethodcopy_d2h
(&self, src: DevicePtr, dst: &mut [u8])
crates/spark-runtime/src/gpu/mock.rs:114
↓ 230 callersMethodarg_f32
Add an f32 argument.
crates/spark-runtime/src/kernel_args.rs:137
↓ 224 callersMethodas_str
Borrow as a string, if it is one.
crates/xgrammar/src/schema/json_value.rs:72
↓ 220 callersMethodalloc
(&self, bytes: usize)
crates/spark-runtime/src/gpu/mock.rs:80
↓ 209 callersMethodfind
Find the representative of `element`'s set, compressing the path. Panics if `element` was never added (matches C++ `XGRAMMAR_CHECK`).
crates/xgrammar/src/fsm/union_find.rs:49
↓ 202 callersMethodget
Get the cached registry (panics if not initialized).
crates/atlas-core/src/registry.rs:201
↓ 194 callersFunctiontry_kernel
Try to load an optional kernel, logging at debug level if it's not found. Returns `KernelHandle(0)` (null) on failure — callers must check before use.
crates/spark-model/src/layers/mod.rs:42
↓ 187 callersFunctiondense
(store: &WeightStore, name: &str)
crates/spark-model/src/weight_map/model_a.rs:121
↓ 174 callersMethodlen
(&self)
crates/spark-runtime/src/radix_tree/snapshot.rs:169
↓ 160 callersMethodfree
Return a snapshot slot to the free list.
crates/spark-model/src/model/ssm_snapshot.rs:362
↓ 155 callersFunctiontest
(name: str, condition: bool, detail: str = "")
scripts/test_coherence.py:48
↓ 153 callersMethodexpect
(&mut self, c: char)
crates/xgrammar/src/schema/json_value.rs:134
↓ 146 callersFunctionfrom_f32
(v: f32)
crates/spark-model/src/weight_loader/deepseek_v4/compute.rs:32
↓ 134 callersMethodinsert
(&mut self, c: LqerCorrection)
crates/spark-server/src/lqer.rs:98
↓ 115 callersMethodarg_u64
Add a u64 argument.
crates/spark-runtime/src/kernel_args.rs:116
↓ 113 callersMethodlaunch_typed
( &self, func: KernelHandle, grid: [u32; 3], block: [u32; 3], _shared_
crates/spark-runtime/src/metal_backend.rs:414
↓ 106 callersMethodlen
Number of elements in the payload.
crates/xgrammar/src/grammar/expr.rs:66
↓ 99 callersMethodcopy_d2d_async
( &self, src: DevicePtr, dst: DevicePtr, bytes: usize, stream: u64,
crates/spark-runtime/src/metal_backend.rs:365
↓ 99 callersMethoddefault_stream
(&self)
crates/spark-runtime/src/gpu/mock.rs:148
↓ 97 callersMethodfree
(&self, ptr: DevicePtr)
crates/spark-runtime/src/gpu/mock.rs:100
↓ 89 callersFunctionrms_norm
RMS normalization: output = rms_norm(input) * weight. Kernel: `rms_norm(input, weight, output, hidden_size, eps)` Grid: (num_tokens, 1, 1) Block: (m
crates/spark-model/src/layers/ops/norm.rs:22
↓ 84 callersFunctionmatcher
A matcher for an EBNF grammar compiled against [`tok`].
crates/xgrammar/src/matcher/tests.rs:69
↓ 81 callersMethodparse
(s: &str)
crates/atlas-core/src/compute.rs:40
↓ 79 callersFunctionbf16_slice_to_bytes
(values: &[half::bf16])
crates/spark-runtime/src/metal_backend/tests/helpers.rs:39
↓ 79 callersMethodcopy_h2d_async
Async host-to-device copy (no stream synchronization). Lifetime requirement**: the source buffer must remain valid until the copy completes (i.e., un
crates/spark-runtime/src/gpu.rs:186
↓ 76 callersFunctiongpu_alloc_zeroed
Allocate `bytes` of GPU memory, zero-initialized.
crates/atlas-spark-bench/src/gpu.rs:40
↓ 76 callersFunctiongpu_free
Free GPU memory.
crates/atlas-spark-bench/src/gpu.rs:54
↓ 75 callersMethodadd_edge
Add a character-range edge `from -[min,max]-> to`.
crates/xgrammar/src/fsm/fsm.rs:93
↓ 72 callersFunctionno_space
Options helper: `any_whitespace = false`, strict.
crates/xgrammar/src/schema/tests.rs:47
↓ 71 callersMethodis_none
(&self)
crates/spark-model/src/layers/mod.rs:62
↓ 70 callersMethoddata
Borrow the underlying parsed AST.
crates/xgrammar/src/api/grammar.rs:91
↓ 69 callersFunctionjson_schema_to_ebnf
Convert a JSON Schema *string* into an EBNF grammar string. Port of the `std::string` overload of `JSONSchemaToEBNF`.
crates/xgrammar/src/schema/api.rs:29
↓ 69 callersMethodlen
Number of stored elements.
crates/xgrammar/src/support/union_find.rs:51
↓ 66 callersMethodlen
Number of rows.
crates/xgrammar/src/fsm/compact_array.rs:36
↓ 66 callersFunctionquantize_to_nvfp4
Quantize a BF16 dense weight to NVFP4 on GPU. Two-phase: (1) find global max, (2) per-group E2M1 quantization. Halves weight bandwidth vs FP8 (0.5 by
crates/spark-model/src/weight_map/loaders_fp8.rs:144
↓ 65 callersMethodpeek
Byte at `pos + delta`, or `0` (NUL) past the end — mirrors the C++ NUL-terminated-string `Peek`.
crates/xgrammar/src/grammar/lexer.rs:62
↓ 64 callersMethodnorm_output
(&self)
crates/spark-runtime/src/buffers.rs:226
↓ 63 callersMethodread
(&mut self, requests: &[ReadRequest], stream: u64)
crates/spark-storage/src/backend/posix.rs:33
↓ 62 callersMethodinsert
Insert or replace. Bumps to the back of the LRU and persists to the backend (no-op when persistence is disabled). If a capacity eviction triggers, the
crates/spark-server/src/response_store/store_impl.rs:117
↓ 61 callersFunctiondense_auto
Load a dense weight, auto-detecting FP8 block-scaled vs BF16/FP32. If the tensor is FP8E4M3 and a `{name_without_.weight}.weight_scale_inv` key exist
crates/spark-model/src/weight_map/quant_helpers.rs:284
↓ 61 callersFunctiondense_gemm
( gpu: &dyn GpuBackend, kernel: KernelHandle, input: DevicePtr, weight: &DenseWeight, outp
crates/spark-model/src/layers/ops/gemm_dense.rs:90
↓ 61 callersMethodexpr
Borrowed view of expression `expr_id`. The CSR slot is `[type, data_len, data0, …]`; this returns `(type, &data[..])`.
crates/xgrammar/src/grammar/data.rs:189
↓ 61 callersMethodtake
(&mut self)
crates/spark-model/src/model/ssm_pool.rs:438
↓ 60 callersMethodlen
Number of loaded weights.
crates/spark-runtime/src/weights.rs:121
↓ 59 callersFunctiondiag_norm
Debug: read back BF16 GPU tensor and compute L2 norm + first 4 values.
crates/spark-model/src/layers/qwen3_attention/trait_impl.rs:18
↓ 58 callersMethodrun
Run the pipeline. Returns (content_before_calls, calls).
crates/spark-server/src/tool_parser/pipeline.rs:80
↓ 57 callersMethodas_array
Borrow as an array slice, if it is one.
crates/xgrammar/src/schema/json_value.rs:64
↓ 57 callersMethodencode
(&self, text: &str)
crates/spark-server/src/tokenizer/chat_impl.rs:74
↓ 56 callersMethodtranspose_for_gemm
Transpose weight layout from [N, K/2] to [K/2, N] for coalesced GEMM reads. Also transposes scale from [N, K/GROUP_SIZE] to [K/GROUP_SIZE, N]. Return
crates/spark-model/src/weight_map/quantized.rs:103
↓ 55 callersMethoddecode
(&self, ids: &[u32])
crates/spark-server/src/tokenizer/chat_impl.rs:82
↓ 55 callersMethodsynchronize
(&self, _stream: u64)
crates/spark-runtime/src/gpu/mock.rs:144
↓ 54 callersFunctionparse_tool_calls
Parse tool calls from completed model output. Scans for `<tool_call></tool_call>` tags and auto-detects inner format (JSON for hermes, XML for qwen3_
crates/spark-server/src/tool_parser/parse_dispatch.rs:37
↓ 54 callersMethodpop
Pop the next state to process, or `None` when drained.
crates/xgrammar/src/earley/queue.rs:65
↓ 53 callersMethodadd_state
Append a fresh state; returns its id.
crates/xgrammar/src/fsm/fsm.rs:87
↓ 53 callersMethodfsm_mut
Mutable underlying FSM.
crates/xgrammar/src/fsm/with_start_end.rs:47
↓ 53 callersMethodis_null
Whether this weight points to NULL (placeholder).
crates/spark-model/src/weight_map/expert.rs:134
↓ 53 callersMethodmemset_async
(&self, ptr: DevicePtr, value: u8, bytes: usize, _stream: u64)
crates/spark-runtime/src/gpu/mock.rs:164
↓ 53 callersMethodscratch
Scratch buffer for MoE routing + kernel metadata uploads.
crates/spark-runtime/src/buffers.rs:269
↓ 50 callersMethodclear
Free all slots; use between decode steps that don't share residency.
crates/spark-storage/src/scratch_pool.rs:173
↓ 50 callersMethodinsert
Insert blocks into the tree. Skips blocks that already exist. Snapshot storage is handled externally by SsmSnapshotIndex. `matched_tokens` is the pre
crates/spark-runtime/src/radix_tree/inner.rs:222
↓ 50 callersMethodis_empty
Whether the set is empty.
crates/xgrammar/src/support/union_find.rs:56
↓ 50 callersMethodshared_mem
(mut self, bytes: u32)
crates/spark-runtime/src/kernel_args.rs:90
↓ 50 callersMethodview
Borrow this layer as the shape the shared forward expects.
crates/spark-model/examples/metal_qwen35_inference/full_attention.rs:70
↓ 49 callersMethodadd_sequence
Add a `Sequence` expr (concatenation of sub-expr ids).
crates/xgrammar/src/grammar/builder.rs:204
↓ 49 callersMethodk_pool_ptr
Get the full K cache pool pointer for a layer (for paged decode kernel).
crates/spark-runtime/src/kv_cache/paged_impl.rs:347
↓ 49 callersFunctionload
(p)
bench/longcode/compare.py:13
↓ 49 callersMethodmoe_output
(&self)
crates/spark-runtime/src/buffers.rs:244
↓ 49 callersFunctionset
(key: &'static str, val: &str)
crates/spark-server/src/tool_parser/tests/streaming_frag.rs:403
↓ 48 callersMethodv_pool_ptr
Get the full V cache pool pointer for a layer.
crates/spark-runtime/src/kv_cache/paged_impl.rs:352
↓ 47 callersFunctioncuda_check
(status: i32, what: &str)
crates/spark-runtime/src/cutlass/tests/mod.rs:34
↓ 46 callersMethodget
Look up a key in an object.
crates/xgrammar/src/schema/json_value.rs:88
↓ 46 callersMethodlen
Number of cached entries — for tests / introspection.
crates/xgrammar/src/compiler/rule_cache.rs:221
↓ 46 callersFunctionresidual_add
BF16 residual add: `residual[i] += src[i]` (in-place). Kernel: `bf16_residual_add(residual, src, n)` Grid: (ceil(n/256), 1, 1) Block: (256, 1, 1)
crates/spark-model/src/layers/ops/activations.rs:128
↓ 46 callersMethodst
(&mut self)
crates/xgrammar/src/grammar/functor/structure_normalizer.rs:49
↓ 46 callersFunctiontest_vocab
Build a minimal vocabulary for testing. Contains basic ASCII + JSON structural tokens.
crates/spark-server/src/grammar/tests/mod.rs:22
↓ 45 callersFunctiondense_gemv
Dense BF16 GEMV (M=1): C = A @ B^T for single-row activations. A: [1, K] BF16, B: [N, K] BF16, C: [1, N] BF16. 8 outputs/block, 32 threads (1 warp) p
crates/spark-model/src/layers/ops/gemm_quant.rs:112
↓ 45 callersFunctionparse
(src: &str)
crates/xgrammar/src/grammar/parser_tests.rs:17
↓ 45 callersMethodsynchronize
(&self, stream: u64)
crates/spark-model/src/model/trait_impl/mod.rs:414
↓ 44 callersMethodadd_choices
Add a `Choices` expr (alternation of sub-expr ids).
crates/xgrammar/src/grammar/builder.rs:209
↓ 44 callersMethoddefault_stream
(&self)
crates/spark-model/src/model/trait_impl/mod.rs:399
↓ 43 callersMethodrule
The rule with the given id. Panics if out of bounds (matches the C++ `XGRAMMAR_DCHECK` contract).
crates/xgrammar/src/grammar/data.rs:142
next →1–100 of 7,049, ranked by callers