Functions — github.com/buyukakyuz/rig

hub / github.com/buyukakyuz/rig / functions

Functions938 in github.com/buyukakyuz/rig

⨍ Functions 938◇ Types & classes 217

↓ 1 callersMethodcos

(&self)

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

↓ 1 callersFunctioncreate_benchmark_activation

(request_id: RequestId, tokens: &[u32])

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

↓ 1 callersFunctioncreate_decode_activation

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

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

↓ 1 callersFunctioncreate_device

(device_str: &str)

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

↓ 1 callersFunctioncreate_initial_activation

( request: &InferenceRequest, tokenizer: &dyn Tokenizer, )

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

↓ 1 callersMethodcreate_pipeline

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

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

↓ 1 callersMethoddecode_tokens_per_second

(&self)

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

↓ 1 callersFunctiondefault_model_version

()

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

↓ 1 callersMethodderegister

(&mut self)

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

↓ 1 callersMethoddevice

(&self)

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

↓ 1 callersFunctiondevice_name

(device: &Device)

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

↓ 1 callersMethodend_generation_session

(&self, request_id: RequestId)

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

↓ 1 callersMethodestablish_peer_connections

( &mut self, assignment: &Assignment, )

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

↓ 1 callersFunctionestimate_gguf_memory

( config: &GgufConfig, spec: &PartitionSpec, is_first: bool, is_last: bool, )

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

↓ 1 callersFunctionestimate_memory

( hidden_size: usize, intermediate_size: usize, vocab_size: usize, spec: &PartitionSpec, i

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

↓ 1 callersFunctionfind_gguf_file

(dir: &std::path::Path)

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

↓ 1 callersFunctionfind_safetensor_files

(model_path: &std::path::Path)

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

↓ 1 callersMethodfirst_address

(&self)

crates/rig-core/src/types/pipeline.rs:159

↓ 1 callersMethodforward_impl

(&self, input: Tensor)

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

↓ 1 callersMethodforward_sample_impl

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

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

↓ 1 callersMethodforward_token

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

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

↓ 1 callersMethodgenerated_tokens

(&self)

crates/rig-coordinator/src/generation.rs:72

↓ 1 callersMethodget_generation_control

( &mut self, request_id: RequestId, )

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

↓ 1 callersMethodget_model_info

(&self, model_id: &ModelId)

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

↓ 1 callersFunctionget_model_name

(path: &Path)

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

↓ 1 callersFunctionget_model_name

(path: &Path)

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

↓ 1 callersMethodget_model_path

(&self, model_id: &ModelId)

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

↓ 1 callersMethodget_node_info

(&self, node_id: NodeId)

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

↓ 1 callersMethodget_nodes_with_model

(&self, model_id: &ModelId)

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

↓ 1 callersMethodget_pending_request

( &mut self, pipeline_id: PipelineId, )

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

↓ 1 callersMethodget_pipeline_last_stage

( &self, pipeline_id: PipelineId, )

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

↓ 1 callersMethodget_sent

(&self)

crates/rig-message-bincode/src/channel.rs:100

↓ 1 callersMethodget_session_finish_data

( &self, request_id: RequestId, )

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

↓ 1 callersMethodgguf_layer_path

(&self, layer_idx: usize, component: &str)

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

↓ 1 callersMethodgguf_weight_names

(&self)

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

↓ 1 callersMethodgguf_weight_names

(&self)

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

↓ 1 callersMethodhandle_cli_create_pipeline

( &self, req: rig_core::CliCreatePipelineRequest, )

crates/rig-coordinator/src/handler.rs:539

↓ 1 callersMethodhandle_cli_create_pipeline_auto

( &self, req: rig_core::CliCreatePipelineAutoRequest, )

crates/rig-coordinator/src/handler.rs:559

↓ 1 callersMethodhandle_cli_get_pipeline

( &self, pipeline_id: rig_core::PipelineId, )

crates/rig-coordinator/src/handler.rs:627

↓ 1 callersMethodhandle_cli_list_pipelines

(&self)

crates/rig-coordinator/src/handler.rs:641

↓ 1 callersMethodhandle_cli_message

(&self, msg: CliMessage)

crates/rig-coordinator/src/handler.rs:161

↓ 1 callersMethodhandle_cli_submit_request

( &self, req: CliSubmitRequest, )

crates/rig-coordinator/src/handler.rs:432

↓ 1 callersMethodhandle_deregister

(&mut self)

crates/rig-coordinator/src/handler.rs:237

↓ 1 callersMethodhandle_generation_started

( &self, request_id: rig_core::RequestId, eos_token: u32, prompt_tokens: usize

crates/rig-coordinator/src/handler.rs:316

↓ 1 callersMethodhandle_get_assignment

(&self)

crates/rig-coordinator/src/handler.rs:222

↓ 1 callersMethodhandle_get_generation_control

( &self, request_id: rig_core::RequestId, )

crates/rig-coordinator/src/handler.rs:285

↓ 1 callersMethodhandle_get_pending_request

( &self, pipeline_id: rig_core::PipelineId, )

crates/rig-coordinator/src/handler.rs:248

↓ 1 callersMethodhandle_get_status

(&self)

crates/rig-coordinator/src/handler.rs:425

↓ 1 callersMethodhandle_heartbeat

( &self, req: rig_core::HeartbeatRequest, )

crates/rig-coordinator/src/handler.rs:207

↓ 1 callersMethodhandle_incoming

( &mut self, msg: CoordinatorIncoming, )

crates/rig-coordinator/src/handler.rs:99

↓ 1 callersMethodhandle_register

( &mut self, req: RegisterRequest, )

crates/rig-coordinator/src/handler.rs:174

↓ 1 callersMethodhandle_report_ready

( &self, pipeline_id: rig_core::PipelineId, )

crates/rig-coordinator/src/handler.rs:228

↓ 1 callersMethodhandle_return_generation_decision

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

crates/rig-coordinator/src/handler.rs:265

↓ 1 callersMethodhandle_streaming_complete

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

crates/rig-coordinator/src/handler.rs:402

↓ 1 callersMethodhandle_streaming_request

( &self, pipeline_id: rig_core::PipelineId, inference_request: InferenceRequest,

crates/rig-coordinator/src/handler.rs:453

↓ 1 callersMethodhandle_token_generated

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

crates/rig-coordinator/src/handler.rs:388

↓ 1 callersMethodhandle_token_sampled

( &self, request_id: rig_core::RequestId, token: u32, )

crates/rig-coordinator/src/handler.rs:339

↓ 1 callersMethodhandle_worker_message

( &mut self, msg: WorkerMessage, )

crates/rig-coordinator/src/handler.rs:115

↓ 1 callersMethodhas_embeddings

(&self)

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

↓ 1 callersMethodinterpret_rope_scaling

(&self, _config: &RopeScalingConfig)

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

↓ 1 callersMethodis_error

(&self)

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

↓ 1 callersMethodis_initialized

(&self)

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

↓ 1 callersMethodis_multi_stage_last

(&self)

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

↓ 1 callersMethodlist_pipelines

(&self)

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

↓ 1 callersMethodlist_pipelines

(&self)

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

↓ 1 callersMethodlisten_str

(&self, addr: &str)

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

↓ 1 callersMethodload

(path: Option<&Path>)

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

↓ 1 callersMethodload_config

( &self, model_path: impl AsRef<std::path::Path>, )

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

↓ 1 callersFunctionload_from_default_locations

()

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

↓ 1 callersFunctionload_gguf_block

( loader: &GgufLoader, layer_idx: usize, config: &GgufConfig, attn_config: AttentionConfig,

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

↓ 1 callersFunctionload_safetensor_block

( loader: &SafetensorLoader, layer_idx: usize, config: &TransformerConfig, attn_config: Attent

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

↓ 1 callersFunctionload_tokenizer_config

(model_path: &Path)

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

↓ 1 callersFunctionmain

()

bench/compare.py:154

↓ 1 callersMethodmark_unhealthy

(&self, node_id: NodeId, reason: &str)

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

↓ 1 callersMethodmax_seq_len

(&self)

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

↓ 1 callersMethodmessage_loop

(&mut self)

crates/rig-coordinator/src/handler.rs:60

↓ 1 callersMethodon_token_sampled

( &self, request_id: RequestId, token: u32, )

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

↓ 1 callersFunctionparse_dtype

(s: &str)

crates/rig-cli/src/commands/pipeline.rs:94

↓ 1 callersFunctionparse_input

(args: &GenerateArgs)

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

↓ 1 callersFunctionparse_model_spec

(spec: &str)

crates/rig-cli/src/commands/worker.rs:41

↓ 1 callersFunctionparse_prompt_sizes

(s: &str)

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

↓ 1 callersFunctionparse_stage_spec

(spec: &str)

crates/rig-cli/src/commands/pipeline.rs:75

↓ 1 callersMethodpartition

(&self)

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

↓ 1 callersMethodpeer_listen_addr

(&self)

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

↓ 1 callersMethodpending_request_count

(&self, pipeline_id: PipelineId)

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

↓ 1 callersMethodprefill_tokens_per_second

(&self)

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

↓ 1 callersFunctionquery_cuda_memory

()

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

↓ 1 callersFunctionquery_metal_memory

()

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

↓ 1 callersMethodrecv_activation

(&self)

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

↓ 1 callersMethodregister_multi_stage_request

( &self, request_id: RequestId, )

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

↓ 1 callersMethodrelease_request_cache

(&self, _request_id: rig_core::RequestId)

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

↓ 1 callersMethodrelease_request_cache

(&self, _request_id: RequestId)

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

↓ 1 callersMethodrun

(&mut self, model_id: ModelId)

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

↓ 1 callersFunctionrun_benchmark

(args: BenchmarkArgs)

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

↓ 1 callersFunctionrun_config

(args: &ConfigArgs)

crates/rig-cli/src/commands/config.rs:28

↓ 1 callersFunctionrun_coordinator

(args: CoordinatorArgs, config: &RigConfig)

crates/rig-cli/src/commands/coordinator.rs:29

↓ 1 callersFunctionrun_demo

(args: DemoArgs, config: &RigConfig)

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

↓ 1 callersMethodrun_first_stage_loop

( &mut self, assignment: &Assignment, mut shutdown_rx: broadcast::Receiver<()>, )

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

↓ 1 callersFunctionrun_generate

(args: GenerateArgs, config: &RigConfig)

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

↓ 1 callersFunctionrun_heartbeat_loop

( node_id: NodeId, mut client: CoordinatorClient<T, C>, interval: Duration, shutdown_rx: &mut

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

← previous next →201–300 of 938, ranked by callers