MCPcopy Create free account

hub / github.com/FastMAS/KVCOMM / functions

Functions446 in github.com/FastMAS/KVCOMM

↓ 2 callersMethod_rotate_segment_caches
(self, segment_meta: Dict[str, Any])
KVCOMM/llm/kvcomm_engine.py:705
↓ 2 callersFunction_slice_inplace
In-place slice of KV cache along sequence dimension.
KVCOMM/llm/kvcomm_engine.py:224
↓ 2 callersFunction_strip_string
(s)
datasets/gsm8k_dataset.py:23
↓ 2 callersMethodagen_kvcomm
Core KV-aware generation entry. Builds merged prefix KV and token ids from stored segments and per-request placeholder caches, then r
KVCOMM/llm/gpt_chat.py:1161
↓ 2 callersMethodcheck_cycle
Detect if adding edges would create a cycle starting at `new_node`.
KVCOMM/graph/graph.py:430
↓ 2 callersFunctionconcat
Returns a new TokenInput instance with token inputs concatenated along the time dimension.
KVCOMM/llm/token_ops.py:104
↓ 2 callersFunctiondownload
()
datasets/MMLU/download.py:7
↓ 2 callersMethodfetch_shared_cache
Retrieve shared KV cache and ids for a placeholder given message context.
KVCOMM/llm/kvcomm_engine.py:1215
↓ 2 callersFunctionfind_strings_between_pluses
(text)
KVCOMM/agents/analyze_agent.py:17
↓ 2 callersMethodget_analyze_constraint
Return role-specific analysis constraints.
KVCOMM/prompt/mmlu_prompt_set.py:108
↓ 2 callersMethodget_answer_prompt
TODO
KVCOMM/prompt/prompt_set.py:26
↓ 2 callersMethodget_constraint
TODO
KVCOMM/prompt/prompt_set.py:16
↓ 2 callersFunctionget_wikipedia_summary
(title)
KVCOMM/tools/search/wiki.py:39
↓ 2 callersMethodparse_file
(file_path: Path)
KVCOMM/tools/reader/readers.py:57
↓ 2 callersMethodpostprocess_answer
Normalize answer into a single string and strip prefix markers.
KVCOMM/prompt/mmlu_prompt_set.py:174
↓ 2 callersMethodprint
(self)
experiments/accuracy.py:16
↓ 2 callersMethodread_file
(self, file_path: Path, task="describe the file")
KVCOMM/tools/reader/readers.py:311
↓ 2 callersMethodrecord_agent_output
Log an agent's generation output and update reuse statistics.
KVCOMM/utils/metrics.py:51
↓ 2 callersMethodrecord_to_input
(record: pd.DataFrame)
datasets/mmlu_dataset.py:60
↓ 2 callersMethodrun
(self)
KVCOMM/tools/coding/executor_utils.py:16
↓ 2 callersMethodsearch
(self, query)
KVCOMM/tools/search/wiki.py:12
↓ 2 callersMethodstart_request
Initialise bookkeeping for a new request.
KVCOMM/utils/metrics.py:32
↓ 2 callersMethodtrim_token_ids
(ids_dict: Dict[str, torch.Tensor], drop_num: int)
KVCOMM/llm/kvcomm_engine.py:1260
↓ 1 callersMethod__getstate__
(self)
KVCOMM/llm/gpt_chat.py:1795
↓ 1 callersMethod__getstate__
(self)
KVCOMM/graph/graph.py:443
↓ 1 callersMethod__setstate__
(self, state)
KVCOMM/llm/gpt_chat.py:1804
↓ 1 callersMethod__setstate__
(self, state)
KVCOMM/graph/graph.py:448
↓ 1 callersMethod_commit_staged_states_locked
Commit all staged request states into the global dictionaries.
KVCOMM/llm/kvcomm_engine.py:753
↓ 1 callersMethod_compute_anchor_weight_entry
Compute attention-like weights between the real placeholder segment and stored anchors.
KVCOMM/llm/kvcomm_engine.py:814
↓ 1 callersFunction_concat_functional
Return a new cache that is the concatenation of base and others.
KVCOMM/llm/kvcomm_engine.py:346
↓ 1 callersMethod_ensure_global_input_buckets
Ensure the global input buckets exist and return the shared store.
KVCOMM/llm/gpt_chat.py:545
↓ 1 callersMethod_ensure_local
(self, key: str)
KVCOMM/llm/kvcomm_engine.py:545
↓ 1 callersMethod_ensure_thread_pool
Initialise or resize the shared thread pool used for CPU work.
KVCOMM/llm/gpt_chat.py:525
↓ 1 callersMethod_execute
To be overriden by the descendant class
KVCOMM/graph/node.py:262
↓ 1 callersMethod_extract_assistant_prompt
Extract trailing assistant prompt from a legacy text prefix.
KVCOMM/llm/gpt_chat.py:294
↓ 1 callersMethod_extract_chat_markers
Parse tokenizer chat template to identify structural markers.
KVCOMM/llm/gpt_chat.py:239
↓ 1 callersMethod_extract_python_code
Extract pure python code from a fenced block.
KVCOMM/agents/final_decision.py:57
↓ 1 callersFunction_fix_a_slash_b
(string)
datasets/gsm8k_dataset.py:182
↓ 1 callersFunction_fix_fracs
(string)
datasets/gsm8k_dataset.py:151
↓ 1 callersFunction_fix_sqrt
(string)
datasets/gsm8k_dataset.py:123
↓ 1 callersMethod_get_cached_anchor_weights
Look up cached anchor weights for a specific placeholder/message/signature.
KVCOMM/llm/kvcomm_engine.py:774
↓ 1 callersMethod_get_request_state
Return or create a request-scoped state container under a lock.
KVCOMM/llm/kvcomm_engine.py:718
↓ 1 callersMethod_initialize_shared_resources
Lazy-load shared tokenizer/model and shared KV memory storage.
KVCOMM/llm/gpt_chat.py:929
↓ 1 callersFunction_install_dynamic_cache_extensions
Monkey-patch DynamicCache with convenience methods used by KVCOMM.
KVCOMM/llm/kvcomm_engine.py:505
↓ 1 callersMethod_legacy_prompt_from_messages
Fallback prompt renderer when chat_template is unavailable.
KVCOMM/llm/gpt_chat.py:378
↓ 1 callersMethod_load_data
( data_path: str, )
datasets/mmlu_dataset.py:23
↓ 1 callersFunction_make_random_token_sequence
(length: int)
experiments/benchmark_TTFT.py:81
↓ 1 callersMethod_normalise_messages
Convert mixed message representations into chat dicts.
KVCOMM/llm/gpt_chat.py:325
↓ 1 callersMethod_placeholder_length
(cache: DynamicCache)
KVCOMM/llm/kvcomm_engine.py:702
↓ 1 callersMethod_prepare_prefix_template
Normalise various prefix formats into a base messages template.
KVCOMM/llm/gpt_chat.py:276
↓ 1 callersMethod_process_inputs
To be overriden by the descendant class
KVCOMM/graph/node.py:272
↓ 1 callersFunction_remove_right_units
(string)
datasets/gsm8k_dataset.py:196
↓ 1 callersFunction_replace_functional
Return a copy of cache with [start, end) replaced by `real`.
KVCOMM/llm/kvcomm_engine.py:367
↓ 1 callersFunction_resolve_latency_path
(target: Optional[Union[str, Path]])
KVCOMM/llm/gpt_chat.py:54
↓ 1 callersMethod_rotate_half
(x: torch.Tensor)
KVCOMM/llm/kvcomm_engine.py:1173
↓ 1 callersMethod_select_anchor_indices
Pick anchors whose placeholder span covers the current placeholder length.
KVCOMM/llm/kvcomm_engine.py:806
↓ 1 callersFunction_select_indices
Select positions by index tensor, preserving layout and metadata.
KVCOMM/llm/kvcomm_engine.py:373
↓ 1 callersMethod_set_cached_anchor_weights
Store computed anchor weights for reuse within the same request.
KVCOMM/llm/kvcomm_engine.py:793
↓ 1 callersFunction_slice_functional
Return a sliced copy of the KV cache.
KVCOMM/llm/kvcomm_engine.py:266
↓ 1 callersFunction_split_cache
Split a cache into multiple segments by the given lengths (sum sizes).
KVCOMM/llm/kvcomm_engine.py:495
↓ 1 callersFunction_split_cache_by_placeholders
Split a cache into placeholder and prefix segments per provided spans.
KVCOMM/llm/kvcomm_engine.py:410
↓ 1 callersFunction_to_device
Move all tensors in the cache to the specified device.
KVCOMM/llm/kvcomm_engine.py:393
↓ 1 callersFunction_write_per_agent_latency
(output_dir: str)
experiments/benchmark_TTFT.py:119
↓ 1 callersFunctionachat
Call an OpenAI-compatible chat endpoint asynchronously.
KVCOMM/llm/gpt_chat.py:106
↓ 1 callersMethodadd_node
(self, node: Node)
KVCOMM/graph/graph.py:128
↓ 1 callersMethodagen_kvcomm_time_test
Core KV-aware generation entry. Builds merged prefix KV and token ids from stored segments and per-request placeholder caches, then r
KVCOMM/llm/gpt_chat.py:1467
↓ 1 callersMethodanchor_signature
Create a lightweight fingerprint for the active anchors.
KVCOMM/llm/kvcomm_engine.py:770
↓ 1 callersMethodclear_spatial_connection
Clear all the spatial connection of the nodes in the graph.
KVCOMM/graph/graph.py:156
↓ 1 callersMethodclear_temporal_connection
Clear all the temporal connection of the nodes in the graph.
KVCOMM/graph/graph.py:167
↓ 1 callersFunctiondataloader
(data_list, batch_size, i_batch)
experiments/run_humaneval.py:42
↓ 1 callersFunctiondataloader
(data_list, batch_size, i_batch)
experiments/run_gsm8k.py:45
↓ 1 callersFunctioneval_loader
(batch_size: int)
experiments/evaluate_mmlu.py:35
↓ 1 callersFunctionevaluate
( graph: Graph, *, samples: int, output_dir: str, **kwargs )
experiments/benchmark_TTFT.py:86
↓ 1 callersFunctionevaluate
Evaluate a graph on the provided dataset.
experiments/evaluate_mmlu.py:17
↓ 1 callersMethodexecute
(self, func: str, tests: List[str], timeout: int = 5)
KVCOMM/tools/coding/executor_types.py:11
↓ 1 callersFunctionextract_answer_from_output
(completion)
KVCOMM/utils/utils.py:12
↓ 1 callersFunctionextract_dimensions
(url)
KVCOMM/tools/vgen/dalle3.py:37
↓ 1 callersMethodextract_example
(self, prompt: str)
KVCOMM/agents/final_decision.py:526
↓ 1 callersFunctionfetch_summary
(session, title)
KVCOMM/tools/search/wiki.py:21
↓ 1 callersMethodformat_chat_segment
Render a single chat block for the given role and content.
KVCOMM/llm/gpt_chat.py:454
↓ 1 callersMethodgen_video
(self, task: Optional[str] = None, video: Optional[str] = None)
KVCOMM/llm/visual_llm.py:13
↓ 1 callersFunctiongenerate_image
(prompt, api_key)
KVCOMM/tools/vgen/dalle3.py:23
↓ 1 callersFunctiongenerate_layered_graph
(n, layer_num=2)
experiments/run_humaneval.py:230
↓ 1 callersFunctiongenerate_layered_graph
(n, layer_num=2)
experiments/run_mmlu.py:153
↓ 1 callersFunctiongenerate_layered_graph
(n, layer_num=2)
experiments/benchmark_TTFT.py:197
↓ 1 callersFunctiongenerate_layered_graph
(n, layer_num=2)
experiments/run_gsm8k.py:247
↓ 1 callersFunctiongenerate_mesh_graph
(n)
experiments/run_mmlu.py:169
↓ 1 callersFunctiongenerate_mesh_graph
(n)
experiments/benchmark_TTFT.py:213
↓ 1 callersFunctiongenerate_star_graph
(n)
experiments/run_humaneval.py:246
↓ 1 callersFunctiongenerate_star_graph
(n)
experiments/run_mmlu.py:176
↓ 1 callersFunctiongenerate_star_graph
(n)
experiments/benchmark_TTFT.py:220
↓ 1 callersFunctiongenerate_star_graph
(n)
experiments/run_gsm8k.py:263
↓ 1 callersMethodgenerate_with_dense_prefill
Generate with dense prefix prefill and optional anchor update.
KVCOMM/llm/gpt_chat.py:1092
↓ 1 callersMethodgenerate_with_kv_reuse
Generate by reusing existing prefix KV (fast path).
KVCOMM/llm/gpt_chat.py:1122
↓ 1 callersMethodget
(cls, name: str, *args, **kwargs)
KVCOMM/prompt/prompt_set_registry.py:18
↓ 1 callersFunctionget_call_str
(assert_statement: str)
KVCOMM/tools/coding/python_executor.py:12
↓ 1 callersFunctionget_kwargs
( mode: Union[ Literal["DirectAnswer"], Literal["FullConnected"], Literal["Random"
experiments/run_humaneval.py:214
↓ 1 callersFunctionget_kwargs
( mode: Union[ Literal["DirectAnswer"], Literal["FullConnected"], Literal["Random"
experiments/run_mmlu.py:136
↓ 1 callersFunctionget_kwargs
( mode: Union[ Literal["DirectAnswer"], Literal["FullConnected"], Literal["Random"
experiments/benchmark_TTFT.py:180
↓ 1 callersFunctionget_kwargs
( mode: Union[ Literal["DirectAnswer"], Literal["FullConnected"], Literal["Random"
experiments/run_gsm8k.py:231
← previousnext →101–200 of 446, ranked by callers