MCPcopy Create free account

hub / github.com/buyukakyuz/rig / functions

Functions938 in github.com/buyukakyuz/rig

Methodprompt_tokens
(&self)
crates/rig-coordinator/src/generation.rs:87
Methodqmatmul
(&self)
crates/rig-runtime-candle/src/weights/quantized.rs:25
Methodread_exact_bytes
( reader: &mut ReadHalf<TcpStream>, n: usize, )
crates/rig-transport-tcp/src/transport.rs:40
Methodrecv_activation
(&self)
crates/rig-worker/src/peer_connection.rs:51
Functionrecv_receives_and_decodes
()
crates/rig-message-bincode/src/channel.rs:147
Functionrecv_returns_error_on_empty
()
crates/rig-message-bincode/src/channel.rs:163
Methodregister
(&mut self, info: NodeInfo)
crates/rig-worker/src/coordinator_client.rs:51
Functionregister_request_creation
()
crates/rig-core/src/types/protocol.rs:514
Functionregister_response_accepted
()
crates/rig-core/src/types/protocol.rs:521
Functionregister_response_rejected
()
crates/rig-core/src/types/protocol.rs:530
Methodrejected
(reason: impl Into<String>)
crates/rig-core/src/types/protocol.rs:150
Methodrejected
(reason: impl Into<String>)
crates/rig-core/src/types/request.rs:479
Functionrejects_oversized_message
()
crates/rig-transport-tcp/src/transport.rs:204
Methodremaining_capacity
(&self)
crates/rig-core/src/types/tensor.rs:251
Methodreport_ready
(&mut self, pipeline_id: PipelineId)
crates/rig-worker/src/coordinator_client.rs:121
Methodrequest_id
(&self)
crates/rig-coordinator/src/generation.rs:82
Functionrequest_id_generates_unique
()
crates/rig-core/src/types/id.rs:213
Functionrequest_id_serialization_roundtrip
()
crates/rig-core/src/types/id.rs:229
Methodrun
( &mut self, mut shutdown_rx: broadcast::Receiver<()>, )
crates/rig-worker/src/stage.rs:223
Methodrun
(mut self)
crates/rig-coordinator/src/handler.rs:44
Functionrun_interactive_chat
( coordinator_addr: SocketAddr, pipeline_id: PipelineId, config: &GenerationConfig, )
crates/rig-cli/src/commands/demo.rs:270
Functionruntime_default_values
()
crates/rig-cli/src/config.rs:466
Functionruntime_serde_roundtrip
()
crates/rig-cli/src/config.rs:473
Functionruntime_type_display
()
crates/rig-core/src/types/node.rs:282
Methodsampling_params
(&self)
crates/rig-coordinator/src/generation.rs:92
Methodselect_device
()
crates/rig-runtime-candle/src/runtime.rs:82
Methodsend_activation
(&self, activation: &Activation)
crates/rig-worker/src/peer_connection.rs:41
Functionsend_and_receive_empty_message
()
crates/rig-transport-tcp/src/transport.rs:175
Functionsend_and_receive_small_message
()
crates/rig-transport-tcp/src/transport.rs:146
Functionsend_encodes_and_transmits
()
crates/rig-message-bincode/src/channel.rs:127
Methodsend_generation_decision
( &mut self, request_id: RequestId, decision: GenerationDecision, )
crates/rig-worker/src/coordinator_client.rs:221
Functionsend_receive_roundtrip
()
crates/rig-transport-tcp/tests/integration.rs:9
Functionserde_roundtrip
()
crates/rig-cli/src/config.rs:345
Methodserialization
(msg: impl Into<String>)
crates/rig-worker/src/error.rs:65
Functionserialization_roundtrip
()
crates/rig-core/src/types/partitioning.rs:399
Functionserialize_deserialize_roundtrip
()
crates/rig-worker/src/peer_connection.rs:275
Methodset_prev_peer
(&mut self, peer: PeerConnection<T>)
crates/rig-worker/src/stage.rs:105
Functionshape_empty
()
crates/rig-core/src/types/tensor.rs:281
Functionshape_from_vec
()
crates/rig-core/src/types/tensor.rs:296
Functionshape_num_elements
()
crates/rig-core/src/types/tensor.rs:273
Functionshape_size_bytes
()
crates/rig-core/src/types/tensor.rs:288
Methodshould_stop_with_text
( &self, generated_tokens: &[u32], decoded_text: &str, )
crates/rig-core/src/types/stop.rs:81
Methodshutdown
(&self)
crates/rig-worker/src/node.rs:84
Methodshutdown
(&self)
crates/rig-coordinator/src/server.rs:50
Methodsize_bytes
(&self, dtype: DType)
crates/rig-core/src/types/tensor.rs:58
Methodspec
(&self)
crates/rig-runtime-candle/src/partition/mod.rs:499
Methodstage
(&self)
crates/rig-worker/src/node.rs:274
Functionstage_id_display
()
crates/rig-core/src/types/id.rs:253
Methodstage_mut
(&mut self)
crates/rig-worker/src/node.rs:279
Methodstart
(model_name: &str, model_path: &Path, args: &InferenceArgs)
crates/rig-cli/src/commands/inference.rs:134
Methodstart
( model_name: &str, model_path: &Path, args: &DemoArgs, config: &RigConfig,
crates/rig-cli/src/commands/demo.rs:91
Methodstart
()
crates/rig-worker/tests/integration.rs:30
Methodstate
(&self)
crates/rig-coordinator/src/request.rs:32
Methodstep
( &mut self, token_id: u32, )
crates/rig-runtime-candle/src/tokenizer.rs:24
Methodstop_sequences
(&self)
crates/rig-core/src/types/stop.rs:115
Methodsuccess
(node_id: NodeId)
crates/rig-core/src/types/request.rs:474
Methodsuccess_text
(request_id: RequestId, text: impl Into<String>, usage: UsageStats)
crates/rig-core/src/types/request.rs:367
Methodsupports_chat_template
(&self)
crates/rig-runtime-candle/src/tokenizer.rs:148
Methodsupports_chat_template
(&self)
crates/rig-core/src/traits/tokenizer.rs:24
Methodsupports_dtype
(&self, dtype: DType)
crates/rig-core/src/types/node.rs:104
Methodsystem
(content: impl Into<String>)
crates/rig-core/src/types/request.rs:78
Methodtcp
(addr: SocketAddr)
crates/rig-core/src/types/node.rs:187
Methodtcp_from_host_port
(host: &str, port: u16)
crates/rig-core/src/types/node.rs:197
Functiontenant_id_display
()
crates/rig-core/src/types/id.rs:260
Methodtensor_cache
(&self)
crates/rig-runtime-candle/src/kv_cache.rs:31
Functiontensor_data_cpu
()
crates/rig-core/src/types/tensor.rs:302
Functiontest_allocate_and_release
()
crates/rig-runtime-candle/src/kv_cache.rs:108
Functiontest_attention_config_constants
()
crates/rig-runtime-candle/src/model/mod.rs:314
Functiontest_checker_accessors
()
crates/rig-core/src/types/stop.rs:194
Functiontest_config_parsing
()
crates/rig-runtime-candle/tests/integration.rs:34
Functiontest_coordinator_config_defaults
()
crates/rig-coordinator/tests/integration.rs:235
Functiontest_cpu_memory_query
()
crates/rig-runtime-candle/src/memory.rs:53
Functiontest_create_pipeline
()
crates/rig-coordinator/src/state.rs:872
Functiontest_default_values
()
crates/rig-runtime-candle/src/config.rs:228
Functiontest_deregister_node
()
crates/rig-coordinator/src/state.rs:911
Functiontest_detect_architecture_config_llama
()
crates/rig-runtime-candle/src/model/mod.rs:349
Functiontest_detect_architecture_config_qwen
()
crates/rig-runtime-candle/src/model/mod.rs:387
Functiontest_detect_architecture_gguf_llama
()
crates/rig-runtime-candle/src/model/mod.rs:328
Functiontest_detect_architecture_gguf_qwen
()
crates/rig-runtime-candle/src/model/mod.rs:365
Functiontest_device_memory_cpu
()
crates/rig-runtime-candle/src/memory.rs:60
Functiontest_dtype_conversion
()
crates/rig-runtime-candle/src/partition/mod.rs:860
Functiontest_duplicate_allocation_fails
()
crates/rig-runtime-candle/src/kv_cache.rs:145
Functiontest_eos_detection
()
crates/rig-core/src/types/stop.rs:163
Functiontest_eos_token_id
()
crates/rig-runtime-candle/src/config.rs:250
Functiontest_find_dead_nodes
()
crates/rig-coordinator/src/state.rs:924
Functiontest_first_partition_embedding
()
crates/rig-runtime-candle/tests/integration.rs:140
Functiontest_forward_decode_single_token
()
crates/rig-runtime-candle/tests/integration.rs:234
Functiontest_forward_sample_on_non_final_partition_returns_none
()
crates/rig-runtime-candle/tests/integration.rs:452
Functiontest_forward_with_token_ids
()
crates/rig-runtime-candle/tests/integration.rs:196
Functiontest_generation_session_creation
()
crates/rig-coordinator/src/generation.rs:128
Functiontest_get_architecture_llama
()
crates/rig-runtime-candle/src/model/mod.rs:245
Functiontest_get_architecture_llama_causal
()
crates/rig-runtime-candle/src/model/mod.rs:253
Functiontest_get_architecture_mistral
()
crates/rig-runtime-candle/src/model/mod.rs:275
Functiontest_get_architecture_mistral_causal
()
crates/rig-runtime-candle/src/model/mod.rs:282
Functiontest_get_architecture_qwen2
()
crates/rig-runtime-candle/src/model/mod.rs:260
Functiontest_get_architecture_qwen2_causal
()
crates/rig-runtime-candle/src/model/mod.rs:268
Functiontest_get_architecture_unknown
()
crates/rig-runtime-candle/src/model/mod.rs:289
Functiontest_get_assignment_no_assignment
()
crates/rig-coordinator/tests/integration.rs:176
Functiontest_get_pipeline_first_stage
()
crates/rig-coordinator/src/state.rs:998
Functiontest_get_pipeline_last_stage
()
crates/rig-coordinator/src/state.rs:1026
← previousnext →701–800 of 938, ranked by callers