Functions — github.com/buyukakyuz/rig

hub / github.com/buyukakyuz/rig / functions

Functions938 in github.com/buyukakyuz/rig

⨍ Functions 938◇ Types & classes 217

↓ 3 callersMethodrelease

(&mut self, request_id: RequestId)

crates/rig-runtime-candle/src/kv_cache.rs:77

↓ 3 callersMethodsend_activation

(&self, activation: &Activation)

crates/rig-worker/src/stage.rs:139

↓ 3 callersMethodsend_token

( &mut self, request_id: RequestId, token_text: String, )

crates/rig-worker/src/coordinator_client.rs:170

↓ 3 callersFunctionserialize_activation

(activation: &Activation)

crates/rig-worker/src/peer_connection.rs:106

↓ 3 callersMethodshutdown_receiver

(&self)

crates/rig-coordinator/src/server.rs:46

↓ 3 callersMethodstate

(&self)

crates/rig-worker/tests/integration.rs:78

↓ 3 callersMethodstep

(&mut self)

crates/rig-worker/src/stage.rs:211

↓ 3 callersMethodstore_generation_decision

( &self, request_id: RequestId, decision: GenerationDecision, )

crates/rig-coordinator/src/state.rs:550

↓ 3 callersMethodupdate

(&mut self, new_k: Tensor, new_v: Tensor)

crates/rig-runtime-candle/src/cache.rs:205

↓ 3 callersMethodwith_chat_template

(mut self, use_chat_template: bool)

crates/rig-core/src/types/request.rs:139

↓ 3 callersMethodwith_connect_timeout

(mut self, timeout: Duration)

crates/rig-transport-tcp/src/config.rs:25

↓ 3 callersMethodwith_coordinator_addr

(mut self, addr: Address)

crates/rig-worker/src/config.rs:27

↓ 3 callersMethodwith_model_paths

(mut self, paths: HashMap<ModelId, PathBuf>)

crates/rig-worker/src/config.rs:51

↓ 3 callersMethodwith_seed

(mut self, seed: u64)

crates/rig-core/src/types/request.rs:145

↓ 3 callersMethodwith_stages

(mut self, num_stages: usize)

crates/rig-core/src/types/protocol.rs:370

↓ 3 callersMethodwith_stop_sequence

(mut self, stop: impl Into<String>)

crates/rig-core/src/types/request.rs:127

↓ 3 callersMethodwith_system_prompt

(mut self, prompt: impl Into<String>)

crates/rig-core/src/types/request.rs:133

↓ 3 callersMethodwith_version

(mut self, version: impl Into<String>)

crates/rig-core/src/types/protocol.rs:358

↓ 2 callersFunctionactivation_to_tensor

( activation: &Activation, dtype: DType, device: &Device, )

crates/rig-runtime-candle/src/utils.rs:54

↓ 2 callersMethodas_tcp

(&self)

crates/rig-core/src/types/node.rs:215

↓ 2 callersMethodclear

(&mut self)

crates/rig-runtime-candle/src/cache.rs:224

↓ 2 callersFunctioncompute_checksum

(data: &[u8])

crates/rig-worker/src/peer_connection.rs:252

↓ 2 callersFunctioncreate_decode_activation

(request_id: RequestId, token: u32, position: u32)

crates/rig-worker/src/node.rs:808

↓ 2 callersMethodcreate_decode_stream

( &self, skip_special_tokens: bool, )

crates/rig-runtime-candle/src/tokenizer.rs:180

↓ 2 callersMethodcreate_pipeline

( &self, config: PipelineConfig, assignments: Vec<(rig_core::NodeId, usize, usize)>,

crates/rig-cli/src/client.rs:103

↓ 2 callersMethodcreate_pipeline

(&mut self, model_name: &str, stages: usize)

crates/rig-cli/src/commands/demo.rs:216

↓ 2 callersMethoddecode_time_ms

(&self)

crates/rig-core/src/types/request.rs:322

↓ 2 callersMethodderegister_node

(&self, node_id: NodeId)

crates/rig-coordinator/src/state.rs:392

↓ 2 callersMethoddiscover_model

(&self, model_id: ModelId, path: &Path)

crates/rig-runtime-candle/src/runtime.rs:126

↓ 2 callersFunctiondtype_to_u8

(dtype: DType)

crates/rig-worker/src/peer_connection.rs:230

↓ 2 callersFunctionextract_token_ids

(activation: &Activation, device: &Device)

crates/rig-runtime-candle/src/utils.rs:108

↓ 2 callersFunctionformat_timestamp

(ts: str)

bench/compare.py:60

↓ 2 callersMethodforward_sample

( &mut self, input: Activation, sampling: &SamplingParams, )

crates/rig-core/src/traits/runtime.rs:34

↓ 2 callersFunctiongenerate_test_tokens

(bos_token: u32, vocab_size: usize, target_length: usize)

crates/rig-cli/src/commands/benchmark.rs:117

↓ 2 callersMethodget

(&self, request_id: RequestId)

crates/rig-runtime-candle/src/kv_cache.rs:69

↓ 2 callersMethodget_pending_request

(&self, pipeline_id: PipelineId)

crates/rig-coordinator/src/state.rs:484

↓ 2 callersMethodget_pipeline_first_stage

( &self, pipeline_id: PipelineId, )

crates/rig-coordinator/src/state.rs:421

↓ 2 callersMethodget_pipeline_info

( &self, pipeline_id: PipelineId, )

crates/rig-coordinator/src/state.rs:745

↓ 2 callersMethodget_status

(&self)

crates/rig-cli/src/client.rs:35

↓ 2 callersMethodgguf_weight_names

(&self)

crates/rig-runtime-candle/src/model/qwen.rs:39

↓ 2 callersMethodheartbeat

(&mut self, status: NodeStatus)

crates/rig-worker/src/coordinator_client.rs:84

↓ 2 callersMethodinit_buffers

( &mut self, batch_size: usize, num_heads: usize, max_seq_len: usize,

crates/rig-runtime-candle/src/cache.rs:159

↓ 2 callersMethodinterpret_rope_scaling

(&self, config: &RopeScalingConfig)

crates/rig-runtime-candle/src/model/llama.rs:43

↓ 2 callersMethodis_online

(&self)

crates/rig-core/src/types/node.rs:174

↓ 2 callersMethodis_pipeline_ready

(&self, pipeline_id: PipelineId)

crates/rig-coordinator/src/state.rs:412

↓ 2 callersMethodlayer_mut

(&mut self, idx: usize)

crates/rig-runtime-candle/src/cache.rs:283

↓ 2 callersFunctionload_benchmark

(path: Path)

bench/compare.py:18

↓ 2 callersMethodload_partition

( &self, assignment: &Assignment, model_spec: &ModelSpec, )

crates/rig-worker/src/node.rs:196

↓ 2 callersMethodnode_count

(&self)

crates/rig-coordinator/src/state.rs:697

↓ 2 callersMethodnum_elements

(&self)

crates/rig-core/src/types/tensor.rs:44

↓ 2 callersMethodpartition_and_tokenizer

(&mut self)

crates/rig-worker/src/stage.rs:86

↓ 2 callersFunctionquery_cpu_memory

()

crates/rig-runtime-candle/src/memory.rs:23

↓ 2 callersMethodregister_with_models

( &mut self, info: NodeInfo, models: Vec<ModelInfo>, )

crates/rig-worker/src/coordinator_client.rs:55

↓ 2 callersMethodreport_ready

(&mut self, assignment: &Assignment)

crates/rig-worker/src/node.rs:257

↓ 2 callersMethodrun

(mut self)

crates/rig-coordinator/src/monitor.rs:32

↓ 2 callersFunctionrun_forward_through_stages

( partitions: &mut [Box<dyn Partition>], input: Activation, sampling: &SamplingParams, )

crates/rig-cli/src/commands/benchmark.rs:219

↓ 2 callersFunctionrun_single_benchmark

( partitions: &mut [Box<dyn Partition>], tokenizer: &dyn Tokenizer, tokens: &[u32], generation

crates/rig-cli/src/commands/benchmark.rs:164

↓ 2 callersMethodsafetensor_weight_names

(&self)

crates/rig-runtime-candle/src/model/qwen.rs:35

↓ 2 callersMethodsend_streaming_complete

( &mut self, request_id: RequestId, usage: UsageStats, )

crates/rig-worker/src/coordinator_client.rs:197

↓ 2 callersMethodseq_len

(&self)

crates/rig-runtime-candle/src/cache.rs:189

↓ 2 callersMethodset_coordinator_client

(&mut self, client: SharedCoordinatorClient<T, C>)

crates/rig-worker/src/stage.rs:109

↓ 2 callersMethodshould_continue

(&self)

crates/rig-core/src/types/stop.rs:13

↓ 2 callersMethodshutdown_receiver

(&self)

crates/rig-worker/src/node.rs:80

↓ 2 callersMethodsubmit_request

( &self, pipeline_id: PipelineId, request: InferenceRequest, )

crates/rig-coordinator/src/request.rs:19

↓ 2 callersMethodtensor_cache_mut

(&mut self)

crates/rig-runtime-candle/src/kv_cache.rs:35

↓ 2 callersFunctiontest_activation

()

crates/rig-worker/src/peer_connection.rs:267

↓ 2 callersFunctiontest_node_info

()

crates/rig-core/src/types/protocol.rs:501

↓ 2 callersFunctiontest_request

(model_id: ModelId)

crates/rig-coordinator/src/request.rs:52

↓ 2 callersMethodto_toml

(&self)

crates/rig-cli/src/config.rs:146

↓ 2 callersFunctionu8_to_dtype

(value: u8)

crates/rig-worker/src/peer_connection.rs:241

↓ 2 callersMethodupdate_seq_len

(&mut self, request_id: &RequestId, new_len: usize)

crates/rig-runtime-candle/src/kv_cache.rs:39

↓ 2 callersMethodwith_max_message_size

(mut self, size: usize)

crates/rig-transport-tcp/src/config.rs:43

↓ 2 callersMethodwith_model_path

(mut self, model_id: ModelId, path: impl Into<PathBuf>)

crates/rig-worker/src/config.rs:45

↓ 2 callersMethodwith_pipeline_id

(mut self, pipeline_id: PipelineId)

crates/rig-core/src/types/protocol.rs:314

↓ 2 callersMethodwith_top_k

(mut self, top_k: usize)

crates/rig-core/src/types/request.rs:121

↓ 1 callersMethodaccept

( &self, peer_node_id: NodeId, )

crates/rig-worker/src/peer_connection.rs:84

↓ 1 callersFunctionaggregate_metrics

(runs: &[SingleRunMetrics])

crates/rig-cli/src/commands/benchmark.rs:261

↓ 1 callersMethodall

()

crates/rig-core/src/types/request.rs:19

↓ 1 callersMethodapply

(&self, tensor: &candle_core::Tensor)

crates/rig-runtime-candle/src/model/mod.rs:23

↓ 1 callersMethodapply_chat_template

( &self, messages: &[rig_core::ChatMessage], add_generation_prompt: bool, )

crates/rig-runtime-candle/src/tokenizer.rs:106

↓ 1 callersMethodapply_env_overrides

(&mut self)

crates/rig-cli/src/config.rs:179

↓ 1 callersMethodassignments_as_ranges

(&self)

crates/rig-core/src/types/protocol.rs:320

↓ 1 callersMethodattention_config

(&self)

crates/rig-runtime-candle/src/model/mistral.rs:27

↓ 1 callersMethodbias

(&self)

crates/rig-runtime-candle/src/weights/quantized.rs:30

↓ 1 callersMethodbos_token

(&self)

crates/rig-runtime-candle/src/tokenizer.rs:98

↓ 1 callersMethodbuild_cluster_status

(&self)

crates/rig-coordinator/src/state.rs:712

↓ 1 callersFunctionbuild_generation_params

(args: &InferenceArgs)

crates/rig-cli/src/commands/inference.rs:290

↓ 1 callersFunctionbuild_generation_params

(args: &GenerateArgs, config: &RigConfig)

crates/rig-cli/src/commands/generate.rs:95

↓ 1 callersMethodcan_accept_work

(&self)

crates/rig-core/src/types/node.rs:169

↓ 1 callersMethodcharacteristics

(&self)

crates/rig-transport-tcp/src/factory.rs:82

↓ 1 callersMethodclear

(&mut self)

crates/rig-runtime-candle/src/kv_cache.rs:45

↓ 1 callersMethodclear_kv_cache

(&self)

crates/rig-runtime-candle/src/partition/mod.rs:479

↓ 1 callersFunctioncompare_benchmarks

(baseline_path: Path, new_path: Path)

bench/compare.py:51

↓ 1 callersMethodcomplete_streaming_session

( &self, request_id: RequestId, usage: UsageStats, )

crates/rig-coordinator/src/state.rs:529

↓ 1 callersFunctioncompute_single_run_metrics

(usage: &UsageStats, run_index: usize)

crates/rig-cli/src/commands/benchmark.rs:248

↓ 1 callersFunctioncompute_summary

(results: &[SequenceLengthResult], total_time_seconds: f64)

crates/rig-cli/src/commands/benchmark.rs:333

↓ 1 callersMethodconnect_to_coordinator

(&mut self)

crates/rig-worker/src/node.rs:103

↓ 1 callersMethodcontains

(&self, name: &str)

crates/rig-runtime-candle/src/weights/gguf.rs:86

↓ 1 callersFunctionconvert_activation

(config: &TransformerConfig)

crates/rig-runtime-candle/src/partition/mod.rs:555

↓ 1 callersFunctionconvert_dtype

(dtype: rig_core::DType)

crates/rig-runtime-candle/src/partition/mod.rs:544

← previous next →101–200 of 938, ranked by callers