MCPcopy Create free account

hub / github.com/PersiaML/PERSIA / functions

Functions720 in github.com/PersiaML/PERSIA

⨍ Functions 720◇ Types & classes 217↳ Endpoints 1
↓ 106 callersMethoditer
Returns a borrowing iterator over its elements.
rust/persia-embedding-holder/src/array_linked_list.rs:834
↓ 67 callersMethodlen
(&self)
rust/persia-embedding-holder/src/eviction_map.rs:108
↓ 52 callersMethodinto_iter
(self)
rust/persia-embedding-holder/src/array_linked_list.rs:1093
↓ 45 callersMethodas_slice
(&self, num_elements: usize)
rust/persia-core/src/cuda/pinned_memory_pool.rs:53
↓ 36 callersMethodinsert
(&mut self, key: K, value: V)
rust/persia-embedding-holder/src/eviction_map.rs:76
↓ 34 callersMethodappend
(&self, line: String)
rust/persia-storage/src/lib.rs:162
↓ 27 callersMethodget
()
rust/persia-core/src/lib.rs:111
↓ 27 callersMethodset
(replica_size: usize, replica_index: usize)
rust/persia-embedding-config/src/lib.rs:195
↓ 19 callersMethodcreate_gauge
(&self, name: &str, help: &str)
rust/persia-metrics/src/lib.rs:115
↓ 12 callersMethodget_client_by_index
( &self, client_index: usize, )
rust/persia-embedding-server/src/embedding_worker_service/mod.rs:243
↓ 11 callersMethodcreate
(&self, p: bool)
rust/persia-storage/src/lib.rs:84
↓ 11 callersMethodsend
(&self, batch_data: &mut PersiaBatch, py: Python)
rust/persia-core/src/utils.rs:88
↓ 10 callersMethodreplica_size
(&self)
rust/persia-embedding-server/src/embedding_worker_service/mod.rs:239
↓ 9 callersMethodload
(&self, src_dir: String)
rust/persia-core/src/lib.rs:307
↓ 8 callersFunctionget_default_logger
r"""Get the default logger. If default logger is not set, init the default by the given name. Arguments: name (str, optional): logger
persia/logger.py:119
↓ 8 callersMethodinit
(self, force: bool = False)
persia/env.py:25
↓ 8 callersMethodto
(self, device: &Option<i32>)
rust/persia-core/src/tensor.rs:258
↓ 7 callersMethodallocate
(&self, size: usize)
rust/persia-core/src/cuda/resource_pool.rs:41
↓ 7 callersMethodembedding_dim
(&self)
rust/persia-embedding-holder/src/emb_entry.rs:124
↓ 7 callersMethodlist
(&self)
rust/persia-storage/src/lib.rs:147
↓ 7 callersMethodremove
Removes the element at the given index and returns it, or `None` if it is empty. The indices of other items are not changed. Indices, which have never
rust/persia-embedding-holder/src/array_linked_list.rs:488
↓ 7 callersFunctionrun_command
(cmd: List[str], env: Optional[dict] = None)
persia/utils.py:55
↓ 7 callersMethodshard
(&self, key: &u64)
rust/persia-embedding-holder/src/lib.rs:91
↓ 6 callersFunction_cast_dlpack2torch_tensor
Convert the DLPack PythonCapsule to torch tensor. Arguments: Tensor (Tensor): tensor wrapper that contains dlpack information. re
persia/ctx.py:40
↓ 6 callersMethodget
()
rust/persia-embedding-server/src/embedding_worker_service/mod.rs:63
↓ 6 callersMethodget_mut
(&mut self, key: &K)
rust/persia-embedding-holder/src/eviction_map.rs:41
↓ 6 callersMethodget_nats_publish_service
( &self, )
rust/persia-core/src/lib.rs:173
↓ 6 callersFunctioninit_module
(super_module: &PyModule, py: Python)
rust/persia-core/src/nats.rs:425
↓ 6 callersMethodis_file
(&self)
rust/persia-storage/src/lib.rs:95
↓ 6 callersFunctionresolve_binary_execute_path
Resolved executable file under persia package root.
persia/utils.py:64
↓ 5 callersFunctioncreate_compact_request
()
rust/others/persia-common-benchmark/benches/serialize_inf_request.rs:70
↓ 5 callersFunctioncreate_request
()
rust/others/persia-common-benchmark/benches/serialize_inf_request.rs:27
↓ 5 callersMethoddelete
(&self)
k8s/src/lib.rs:37
↓ 5 callersMethodget_slot_by_feature_name
(&self, feature_name: &str)
rust/persia-embedding-config/src/lib.rs:593
↓ 5 callersMethodrecv
(&self)
rust/persia-common/src/message_queue.rs:61
↓ 5 callersMethodsign
(&self)
rust/persia-embedding-holder/src/emb_entry.rs:155
↓ 4 callersMethod__init__
Arguments: threadpool_worker_size (int): rpc threadpool worker size. device_id (int, optional): the CUDA device to us
persia/ctx.py:208
↓ 4 callersMethodapply
(&self)
k8s/src/lib.rs:63
↓ 4 callersMethodgen_pod_template
(&self, job_name: &str, namespace: &str)
k8s/src/crd.rs:67
↓ 4 callersMethodget
(&self, key: &K)
rust/persia-embedding-holder/src/eviction_map.rs:34
↓ 4 callersMethodget_first_client
(&self)
rust/persia-core/src/rpc.rs:37
↓ 4 callersFunctionget_world_size
Get the total number of processes.
persia/env.py:98
↓ 4 callersMethodident_web_api
(&self)
rust/others/persia-rpc-macro/src/lib.rs:16
↓ 4 callersMethodident_web_api_compressed
(&self)
rust/others/persia-rpc-macro/src/lib.rs:20
↓ 4 callersMethodinit_nats_publisher
(&mut self, world_size: Option<usize>)
rust/persia-core/src/lib.rs:214
↓ 4 callersMethodload
(&self, req: String)
rust/persia-embedding-server/src/embedding_worker_service/mod.rs:1150
↓ 4 callersFunctionmake_dataloader
( data_filepath: str, batch_size: int = 128, skip_last_batch: bool = False )
examples/src/adult-income/data_generator.py:75
↓ 4 callersMethodnext_of_prev
(&mut self, index: u32, active: bool)
rust/persia-embedding-holder/src/array_linked_list.rs:433
↓ 4 callersMethodprev_of_next
(&mut self, index: u32, active: bool)
rust/persia-embedding-holder/src/array_linked_list.rs:422
↓ 4 callersMethodrecycle
(&self, item: T)
rust/persia-core/src/cuda/resource_pool.rs:46
↓ 4 callersFunctionsign_to_shard_modulo
(sign: u64, replica_size: u64)
rust/persia-embedding-server/src/embedding_worker_service/mod.rs:342
↓ 4 callersMethodupdate
( &self, emb_entry: &mut [f32], grad: &[f32], dim: usize, emb_opt_stat
rust/persia-common/src/optim.rs:193
↓ 4 callersMethodwait_servers_ready
Wait until embedding servers are ready to serve.
persia/ctx.py:887
↓ 3 callersFunction_batch_size_check
Check if batch size is equal to target_batch_size and no larger than to MAX_BATCH_SIZE.
persia/embedding/data.py:57
↓ 3 callersFunctionadd_assign_avx2
(a: &mut [f32], b: &[f32])
rust/persia-simd/src/lib.rs:4
↓ 3 callersMethodclear
(&self)
rust/persia-embedding-holder/src/lib.rs:87
↓ 3 callersMethodconfigure_embedding_parameter_servers
Apply :class:`EmbeddingConfig` to embedding servers. Arguments: embedding_config (EmbeddingConfig): the embedding configuration t
persia/ctx.py:415
↓ 3 callersMethodconnect_indices
(&mut self, prev_index: u32, next_index: u32, active: bool)
rust/persia-embedding-holder/src/array_linked_list.rs:443
↓ 3 callersFunctionconvert_data_ptr2gradient
( ptr: *mut std::os::raw::c_void, shape: [usize; 2], num_elements: usize, is_f16: bool, )
rust/persia-core/src/backward.rs:135
↓ 3 callersFunctionconvert_pyarray_object_to_tensor_impl
convert pyarray object to persia tensor this function will panic when meet some datatype numpy::Datatype can't support
rust/persia-core/src/data.rs:103
↓ 3 callersMethodcreate_counter
( &self, name: &str, help: &str, )
rust/persia-metrics/src/lib.rs:89
↓ 3 callersFunctioncreate_request_smallvec
()
rust/others/persia-common-benchmark/benches/serialize_inf_request.rs:113
↓ 3 callersMethoddim
(&self)
rust/persia-embedding-holder/src/emb_entry.rs:120
↓ 3 callersMethoddump
(&self, dst_dir: String)
rust/persia-core/src/lib.rs:300
↓ 3 callersMethoddump_torch_state_dict
Dump a Pytorch model or optimizer's state dict to the destination directory. Arguments: torch_instance (torch.nn.Module or torch.
persia/ctx.py:555
↓ 3 callersMethodemb
(&self)
rust/persia-embedding-holder/src/emb_entry.rs:128
↓ 3 callersMethodfilter
Filter the log record whose level is greater than the preset log level. Arguments: record (logging.LogRecord): callback function
persia/logger.py:18
↓ 3 callersMethodget_client_by_addr
(&self, embedding_worker_addr: &str)
rust/persia-core/src/rpc.rs:55
↓ 3 callersFunctionget_default_embedding_config
Get default embedding configuration.
persia/embedding/__init__.py:24
↓ 3 callersMethodget_emb_dump_done_file_name
(&self)
rust/persia-model-manager/src/lib.rs:152
↓ 3 callersFunctionget_embedding_worker_services
Get a list of current embedding worker services.
persia/service.py:6
↓ 3 callersMethodget_random_client
(&self)
rust/persia-core/src/rpc.rs:33
↓ 3 callersMethodget_random_client_with_addr
(&self)
rust/persia-core/src/rpc.rs:26
↓ 3 callersMethodget_raw_ptr
(&mut self)
rust/persia-core/src/tensor.rs:131
↓ 3 callersMethodget_status
(&self)
rust/persia-model-manager/src/lib.rs:119
↓ 3 callersMethodget_world_size
(&self, _placeholder: ())
rust/persia-core/src/nats.rs:136
↓ 3 callersMethodinsert_free_element
(&mut self, element: LinkedListNode<T>)
rust/persia-embedding-holder/src/array_linked_list.rs:255
↓ 3 callersMethodlen
Returns the number of elements in the linked array.
rust/persia-embedding-holder/src/array_linked_list.rs:812
↓ 3 callersMethodname
(self)
persia/embedding/data.py:210
↓ 3 callersMethodnumpy
(&self, py: Python)
rust/persia-core/src/forward.rs:221
↓ 3 callersMethodpush_back
Adds an element at the back of the array and returns its index. The indices are returned in a raising order, starting with zero. See the module descri
rust/persia-embedding-holder/src/array_linked_list.rs:317
↓ 3 callersMethodput
(&self, data: Vec<u8>)
rust/persia-core/src/utils.rs:33
↓ 3 callersFunctionrandom_port
(start: int = 10000, end: int = 65535)
test/utils.py:4
↓ 3 callersMethodready_for_serving
(&self)
rust/persia-embedding-server/src/embedding_worker_service/mod.rs:210
↓ 3 callersMethodrequest
(&self, subject: &str, msg: &[u8])
rust/others/persia-nats-client/src/lib.rs:73
↓ 3 callersMethodrequire_space
(&self, dim: usize)
rust/persia-common/src/optim.rs:137
↓ 3 callersMethodsend_data
Send PersiaBatch from data loader to nn worker and embedding worker side. Arguments: persia_batch (PersiaBatch): :class:`PersiaBa
persia/ctx.py:335
↓ 3 callersFunctionset_device
(card_index: i32)
rust/persia-core/src/cuda/mod.rs:20
↓ 3 callersMethodshutdown
(&self)
rust/persia-core/src/rpc.rs:178
↓ 3 callersFunctionstart_deadlock_detection_thread
()
rust/persia-common/src/utils.rs:21
↓ 3 callersMethodsynchronize
(&self)
rust/persia-core/src/cuda/cuda_event_pool.rs:32
↓ 3 callersMethodupdate_rpc_clients
( &self, servers: Vec<String>, ready_for_serving: bool, )
rust/persia-embedding-server/src/embedding_worker_service/mod.rs:300
↓ 2 callersMethod__init__
Arguments: lr(float): learning rate. momentum(float, optional): momentum factor. weight_decay(float, opti
persia/embedding/optim.py:21
↓ 2 callersFunction_dump_config_into_tempfile
(config: dict)
persia/helper.py:45
↓ 2 callersFunction_dump_function_into_tempfile
( func: Callable, )
persia/helper.py:29
↓ 2 callersMethod_enter
Hook when enter the context
persia/ctx.py:245
↓ 2 callersMethod_exit
Hook when exit the context
persia/ctx.py:249
↓ 2 callersFunction_id_type_data_check
Check the type, dimension and dtype for id_type_feature. Arguments: id_type_feature (np.ndarray): id_type_feature array. feature_
persia/embedding/data.py:21
↓ 2 callersFunction_kill_processes
(process_group: List[subprocess.Popen])
persia/helper.py:117
↓ 2 callersFunction_launch_serve
( server_type: str, replica_num: int, env: dict, port: int = 8888 )
persia/helper.py:52
next →1–100 of 720, ranked by callers