Functions — github.com/buyukakyuz/rig

hub / github.com/buyukakyuz/rig / functions

Functions938 in github.com/buyukakyuz/rig

⨍ Functions 938◇ Types & classes 217

↓ 37 callersMethodget

(&self, seq_len: usize, offset: usize)

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

↓ 25 callersMethodrecv_frame

(&self)

crates/rig-transport-tcp/src/transport.rs:97

↓ 25 callersMethodsend_frame

(&self, frame: &[u8])

crates/rig-transport-tcp/src/transport.rs:57

↓ 21 callersFunctionconnect

(addr: SocketAddr)

crates/rig-coordinator/tests/integration.rs:107

↓ 19 callersMethodpp

(&self, prefix: &str)

crates/rig-runtime-candle/src/weights/safetensor.rs:18

↓ 18 callersMethodcompute_ranges

( &self, num_layers: usize, _num_nodes: usize, )

crates/rig-core/src/types/partitioning.rs:53

↓ 18 callersMethodregister_node

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

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

↓ 16 callersMethodto_vec

(&self)

crates/rig-runtime-candle/src/config.rs:60

↓ 14 callersMethodlocal_socket_addr

(&self)

crates/rig-transport-tcp/src/listener.rs:50

↓ 14 callersMethodsend

(&self, msg: &WorkerMessage)

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

↓ 14 callersFunctiontest_node_info

(node_id: NodeId, port: u16)

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

↓ 13 callersMethoddecode

(&self, data: &[u8])

crates/rig-message-bincode/src/lib.rs:27

↓ 13 callersMethodis_empty

(&self)

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

↓ 12 callersMethodaccept_with_socket_addr

( &self, )

crates/rig-transport-tcp/src/listener.rs:38

↓ 12 callersMethodconnect

(&self, addr: &Address)

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

↓ 12 callersMethodencode

(&self, msg: &M)

crates/rig-message-bincode/src/lib.rs:24

↓ 12 callersFunctionget_architecture

(name: &str)

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

↓ 12 callersFunctionmodel_path

()

crates/rig-runtime-candle/tests/integration.rs:19

↓ 11 callersFunctionmodel_exists

()

crates/rig-runtime-candle/tests/integration.rs:28

↓ 11 callersMethodrecv

(&self)

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

↓ 10 callersMethodload_qmatmul

(&self, name: &str)

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

↓ 10 callersMethodsend

(&self, msg: &M)

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

↓ 10 callersFunctiontest_config

()

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

↓ 9 callersMethodallocate

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

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

↓ 9 callersMethodis_registered

(&self)

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

↓ 9 callersMethodload_dequantized

(&self, name: &str)

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

↓ 9 callersMethodwith_listen_addr

(mut self, addr: SocketAddr)

crates/rig-coordinator/src/config.rs:25

↓ 8 callersMethodconnect

(addr: SocketAddr)

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

↓ 8 callersMethodcontains

(&self, token_id: u32)

crates/rig-runtime-candle/src/config.rs:68

↓ 8 callersMethodcreate_pipeline

( &self, config: PipelineConfig, stage_assignments: Vec<(NodeId, Range<usize>)>,

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

↓ 8 callersFunctionfmt_with_std

(mean: float, std: float, runs: int)

bench/compare.py:131

↓ 8 callersFunctionsend_recv

( transport: &T, codec: &BincodeCodec, msg: WorkerMessage, )

crates/rig-coordinator/tests/integration.rs:25

↓ 7 callersMethoddtype

(&self)

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

↓ 7 callersMethodflush

(&mut self)

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

↓ 7 callersMethodforward

(&self, input: &Tensor)

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

↓ 7 callersMethodget_mut

(&mut self, request_id: RequestId)

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

↓ 7 callersFunctionparse_header_bytes

(slice: &[u8])

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

↓ 7 callersMethodrecv_frame

(&self)

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

↓ 7 callersMethodregister

(&mut self)

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

↓ 7 callersMethodsend_frame

(&self, frame: &[u8])

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

↓ 7 callersMethodwith_heartbeat_interval

(mut self, interval: Duration)

crates/rig-coordinator/src/config.rs:31

↓ 7 callersMethodwith_max_tokens

(mut self, max_tokens: usize)

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

↓ 6 callersMethodaddr

(&self)

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

↓ 6 callersFunctionconnect_to_coordinator

(addr: &Address)

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

↓ 6 callersMethodwith_read_timeout

(mut self, timeout: Option<Duration>)

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

↓ 5 callersMethodas_bytes

(&self)

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

↓ 5 callersMethodforward

(&mut self, activation: Activation)

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

↓ 5 callersFunctiongguf_model_path

()

crates/rig-runtime-candle/tests/integration.rs:277

↓ 5 callersMethodlocal_addr

(&self)

crates/rig-transport-tcp/src/listener.rs:85

↓ 5 callersMethodon_token

(&mut self, token: u32)

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

↓ 5 callersMethodrun

(&mut self)

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

↓ 5 callersMethodsend_and_recv

(&self, msg: CliMessage)

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

↓ 5 callersFunctionstart_test_server

()

crates/rig-coordinator/tests/integration.rs:56

↓ 5 callersMethodwith_heartbeat_timeout

(mut self, timeout: Duration)

crates/rig-coordinator/src/config.rs:37

↓ 5 callersMethodwith_max_nodes

(mut self, max: usize)

crates/rig-coordinator/src/config.rs:49

↓ 4 callersMethodaccept

(&self)

crates/rig-transport-tcp/src/listener.rs:72

↓ 4 callersFunctiondeserialize_activation

(frame: &[u8])

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

↓ 4 callersFunctiondetect_model_format

(path: &Path)

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

↓ 4 callersMethoddims

(&self)

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

↓ 4 callersFunctionformat_change

(baseline: float, new: float)

bench/compare.py:23

↓ 4 callersMethodforward

(&self, x: &Tensor)

crates/rig-runtime-candle/src/layers/mlp.rs:27

↓ 4 callersMethodforward_sample

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

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

↓ 4 callersFunctiongguf_model_exists

()

crates/rig-runtime-candle/tests/integration.rs:286

↓ 4 callersMethodis_first_stage

(&self)

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

↓ 4 callersMethodmark_ready

( &self, node_id: NodeId, pipeline_id: PipelineId, )

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

↓ 4 callersMethodnum_kv_heads

(&self)

crates/rig-runtime-candle/src/config.rs:100

↓ 4 callersFunctionquery_device_memory

(device: &Device)

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

↓ 4 callersMethodsend_response

(&self, response: &CoordinatorOutgoing)

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

↓ 4 callersMethodshould_stop

(&self, generated_tokens: &[u32])

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

↓ 4 callersMethodstate

(&self)

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

↓ 4 callersFunctiontest_node_info

(node_id: NodeId, port: u16)

crates/rig-coordinator/tests/integration.rs:15

↓ 4 callersMethodtokenizer

(&self)

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

↓ 4 callersMethodwith_context

(mut self, context: RequestContext)

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

↓ 4 callersMethodwith_temperature

(mut self, temperature: f32)

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

↓ 4 callersMethodwith_top_p

(mut self, top_p: f32)

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

↓ 3 callersMethodattention_config

(&self)

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

↓ 3 callersMethodbuild

(self)

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

↓ 3 callersMethodcapabilities

(&self)

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

↓ 3 callersMethodcreate_pipeline_auto

( &self, request: CliCreatePipelineAutoRequest, )

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

↓ 3 callersFunctioncreate_runtime

(device: &str)

crates/rig-cli/src/runtime.rs:5

↓ 3 callersFunctiondetect_architecture_config

(config_path: &Path)

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

↓ 3 callersFunctiondetect_architecture_gguf

( content: &candle_core::quantized::gguf_file::Content, )

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

↓ 3 callersMethoddim

(&self, index: usize)

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

↓ 3 callersMethodembed

(&self, tokens: &Tensor)

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

↓ 3 callersMethodeos_token

(&self)

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

↓ 3 callersMethodfind_dead_nodes

(&self, now: Instant, timeout: std::time::Duration)

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

↓ 3 callersFunctionformat_change_inverse

(baseline: float, new: float)

bench/compare.py:37

↓ 3 callersMethodgenerate

( &self, pipeline_id: PipelineId, input: InferenceInput, params: GenerationPar

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

↓ 3 callersMethodget_assignment

(&mut self)

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

↓ 3 callersMethodget_assignment

(&self, node_id: NodeId)

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

↓ 3 callersMethodget_pipeline

(&self, pipeline_id: PipelineId)

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

↓ 3 callersMethodhead_dim

(&self)

crates/rig-runtime-candle/src/config.rs:105

↓ 3 callersMethodheartbeat

(&self, node_id: NodeId, status: NodeStatus)

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

↓ 3 callersMethodinto_parts

(self)

crates/rig-core/src/types/partition.rs:35

↓ 3 callersMethodis_last_stage

(&self)

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

↓ 3 callersMethodis_stopped

(&self)

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

↓ 3 callersMethodlisten

(&self, addr: &Address)

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

↓ 3 callersMethodlocal_addr

(&self)

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

↓ 3 callersFunctionpod_vec_to_bytes

(mut v: Vec<T>)

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

↓ 3 callersMethodrecv

(&self)

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

next →1–100 of 938, ranked by callers