MCPcopy Create free account

hub / github.com/PersiaML/PERSIA / functions

Functions720 in github.com/PersiaML/PERSIA

⨍ Functions 720◇ Types & classes 217↳ Endpoints 1
↓ 1 callersFunctioncheck_pyarray_dtype_valid
(py_object: &PyAny, dtype: &PyAny, name: Option<String>, py: Python)
rust/persia-core/src/data.rs:138
↓ 1 callersFunctioncli
()
persia/k8s_utils.py:7
↓ 1 callersFunctioncli
()
persia/launcher.py:107
↓ 1 callersMethodclient_impl
(&self)
rust/others/persia-rpc-macro/src/lib.rs:152
↓ 1 callersMethodclient_method
(&self, client_field: &Ident)
rust/others/persia-rpc-macro/src/lib.rs:42
↓ 1 callersMethodcmp
(&self, other: &Self)
rust/persia-core/src/data.rs:70
↓ 1 callersFunctioncnt_ctx
Get the :class:`BaseCtx` recently entered.
persia/ctx.py:1067
↓ 1 callersMethodconfigure
( &self, config: PersiaEmbeddingModelHyperparameters, )
rust/persia-embedding-server/src/embedding_parameter_service/mod.rs:440
↓ 1 callersMethodconsume_all_id_type_feature_embedding_tensors
(&mut self)
rust/persia-core/src/forward.rs:274
↓ 1 callersMethodconsume_all_label_tensors
(&mut self)
rust/persia-core/src/forward.rs:281
↓ 1 callersMethodconsume_all_non_id_type_feature_tensors
(&mut self)
rust/persia-core/src/forward.rs:267
↓ 1 callersMethodconsume_cpu_storage
(self)
rust/persia-core/src/tensor.rs:162
↓ 1 callersMethodconsume_dataset
(self)
persia/data.py:178
↓ 1 callersMethodconvert2distributed_model
Arguments: model (torch.nn.Module): the PyTorch model that needs to be converted to data-parallel model. world_size (
persia/distributed.py:127
↓ 1 callersMethodconverted_id_type_features2embedding_tensor
Convert the IdTypeFeatureBatch to EmbeddingTensor.
rust/persia-core/src/data.rs:228
↓ 1 callersFunctioncopy_gradients
( x: &SingleSlotGradient, num_bytes: usize, num_elements: usize, device_id: Arc<Option<i32>>,
rust/persia-core/src/backward.rs:155
↓ 1 callersMethodcreate_gradient_batch
(&mut self)
rust/persia-core/src/forward.rs:299
↓ 1 callersFunctioncuda_d2h
( num_bytes: usize, data_ptr: *mut std::os::raw::c_void, host_ptr: *mut std::os::raw::c_void, )
rust/persia-core/src/cuda/utils.rs:7
↓ 1 callersFunctiondecayed_adagrad_avx2
( adagrad: &mut [f32], embedding: &mut [f32], gradient: &[f32], g_square_momentum: f32, le
rust/persia-simd/src/lib.rs:21
↓ 1 callersFunctiondecayed_adagrad_vectorwise_shared_avx2
remember to update adagrad with gradient squared after this, the function will not do this
rust/persia-simd/src/lib.rs:81
↓ 1 callersFunctiondecayed_sgd_avx2
(emb: &mut [f32], grad: &[f32], wd: f32, lr: f32)
rust/persia-simd/src/lib.rs:124
↓ 1 callersFunctiondelete
(req: web::Json<JobIdentifier>)
k8s/src/bin/server.rs:73
↓ 1 callersMethoddeploy_pods
(&self)
k8s/src/lib.rs:69
↓ 1 callersMethoddeploy_services
(&self)
k8s/src/lib.rs:78
↓ 1 callersFunctiondeserialize_tear_indices
( bytes: &mut [u8], )
rust/others/persia-common-benchmark/benches/serialize_inf_request.rs:316
↓ 1 callersFunctiondetermine_action
(job: &PersiaJob)
k8s/src/bin/operator.rs:104
↓ 1 callersMethoddevice
(&self)
rust/persia-core/src/tensor.rs:286
↓ 1 callersMethoddlpack
(&mut self)
rust/persia-core/src/tensor.rs:314
↓ 1 callersMethoddump_checkpoint
Dump the dense and embedding checkpoint to destination directory. Arguments: dst_dir (str): destination directory. de
persia/ctx.py:1007
↓ 1 callersMethoddump_embedding
( &self, dst_dir: PathBuf, embedding_holder: PersiaEmbeddingHolder, )
rust/persia-model-manager/src/lib.rs:286
↓ 1 callersMethoddump_embedding
Dump embeddings to the destination directory. By default, this function is synchronous and will wait for the completion of embedding l
persia/ctx.py:522
↓ 1 callersMethoddump_embedding_segment
( &self, dst_dir: PathBuf, signs: Vec<u64>, file_index: usize, num_dum
rust/persia-incremental-update-manager/src/lib.rs:178
↓ 1 callersMethoddump_internal_shard_embeddings
( &self, internal_shard_idx: usize, dst_dir: PathBuf, embedding_holder: Persia
rust/persia-model-manager/src/lib.rs:242
↓ 1 callersMethoddump_to_file
( &self, content: &PyBytes, file_dir: String, file_name: String, )
rust/persia-core/src/lib.rs:417
↓ 1 callersFunctioneval_ctx
Get the :class:`EmbeddingCtx` with the :attr:`.EVAL` mode.
persia/ctx.py:1072
↓ 1 callersFunctionexpect_uri
(url: url::Url)
rust/others/persia-rpc/src/lib.rs:44
↓ 1 callersMethodforward_batch_id
( &self, id_type_feature_remote_ref: IDTypeFeatureRemoteRef, )
rust/persia-embedding-server/src/embedding_worker_service/mod.rs:1031
↓ 1 callersMethodgen_pods
(&self, job_name: &str, namespace: &str)
k8s/src/crd.rs:174
↓ 1 callersMethodgen_services
(&self, job_name: &str, namespace: &str)
k8s/src/crd.rs:138
↓ 1 callersFunctiongen_spec
()
k8s/src/bin/e2e.rs:50
↓ 1 callersMethodget
()
rust/persia-incremental-update-manager/src/lib.rs:53
↓ 1 callersMethodget
()
rust/persia-embedding-server/src/embedding_parameter_service/mod.rs:39
↓ 1 callersMethodget_batch
(&self, timeout_ms: u64, py: Python)
rust/persia-core/src/forward.rs:860
↓ 1 callersMethodget_batch_level_state
(&self, signs: &[u64])
rust/persia-common/src/optim.rs:151
↓ 1 callersMethodget_client_by_index
(&self, client_index: usize)
rust/persia-core/src/rpc.rs:46
↓ 1 callersFunctionget_dataloader_pod_name
(job_name: &str, replica_index: usize)
k8s/src/crd.rs:26
↓ 1 callersFunctionget_default_distributed_option
Get default distributed option. Arguments: device_id (int, optional): CUDA device_id. Apply ``backend="nccl"`` to the ``DDPOption``
persia/distributed.py:413
↓ 1 callersFunctionget_default_incremental_dir
()
rust/persia-embedding-config/src/lib.rs:274
↓ 1 callersFunctionget_default_job_name
()
rust/persia-embedding-config/src/lib.rs:278
↓ 1 callersMethodget_dtype
(&self)
rust/persia-core/src/cuda/mod.rs:95
↓ 1 callersMethodget_emb_state
(&self, opt_state: &Option<Vec<f32>>, idx: usize)
rust/persia-common/src/optim.rs:142
↓ 1 callersFunctionget_embedding_dim
()
rust/persia-common/src/optim.rs:338
↓ 1 callersMethodget_embedding_from_bytes
Get embeddings of the serialized input batch data. Arguments: data (PersiaBatch): serialized input data without embeddings.
persia/ctx.py:637
↓ 1 callersFunctionget_embedding_ps_pod_name
(job_name: &str, replica_index: usize)
k8s/src/crd.rs:14
↓ 1 callersMethodget_embedding_worker_addr_list
(&self)
rust/persia-core/src/lib.rs:262
↓ 1 callersFunctionget_embedding_worker_pod_name
(job_name: &str, replica_index: usize)
k8s/src/crd.rs:18
↓ 1 callersFunctionget_grads
()
rust/persia-common/src/optim.rs:314
↓ 1 callersFunctionget_index
(key: &K, count: usize)
rust/persia-embedding-holder/src/sharded.rs:11
↓ 1 callersFunctionget_inference_stub
()
examples/src/adult-income/serve_client.py:20
↓ 1 callersFunctionget_init_embedding
()
rust/persia-common/src/optim.rs:331
↓ 1 callersMethodget_inner
(&self)
rust/persia-core/src/optim.rs:14
↓ 1 callersMethodget_internam_shard_filename
(&self, internal_shard_idx: usize)
rust/persia-model-manager/src/lib.rs:144
↓ 1 callersMethodget_job_type
(&self)
rust/persia-embedding-server/src/embedding_parameter_service/mod.rs:145
↓ 1 callersFunctionget_local_ip
get local ip address by specific socket name or request the 8.8.8.8:80 to retrieve the ip address from response.
rust/persia-embedding-config/src/lib.rs:122
↓ 1 callersFunctionget_logger
r"""Get logger by name. Arguments: name (str): logger name.
persia/logger.py:96
↓ 1 callersMethodget_master_addr
(&self, _placeholder: ())
rust/persia-core/src/nats.rs:42
↓ 1 callersFunctionget_metrics_gateway_pod_name
(job_name: &str)
k8s/src/crd.rs:30
↓ 1 callersMethodget_raw_embedding
(&mut self)
rust/persia-core/src/forward.rs:81
↓ 1 callersMethodget_raw_ptr
(&mut self)
rust/persia-core/src/cuda/mod.rs:90
↓ 1 callersMethodget_receiver
(&self)
rust/persia-core/src/utils.rs:121
↓ 1 callersMethodget_refresh
(&mut self, key: &K)
rust/persia-embedding-holder/src/eviction_map.rs:48
↓ 1 callersMethodget_remote_ref_info
(&self)
rust/persia-core/src/data.rs:23
↓ 1 callersFunctionget_replica_index
Get the replica index of current service. The replica index is a unique identifier assigned to each replica. They are assigned following the
persia/env.py:126
↓ 1 callersFunctionget_replica_size
Get the replica size of the current service. Replica size is the number of services launched by docker service or k8s
persia/env.py:118
↓ 1 callersMethodget_replica_size
(&self)
rust/persia-embedding-server/src/embedding_worker_service/mod.rs:1315
↓ 1 callersMethodget_sender
(&self)
rust/persia-core/src/utils.rs:115
↓ 1 callersMethodget_shard_by_index
( &self, index: usize, )
rust/persia-embedding-holder/src/lib.rs:95
↓ 1 callersFunctionget_stride_by_shape
(shape: &[usize])
rust/persia-core/src/tensor.rs:219
↓ 1 callersMethodget_sum_embedding
(&mut self)
rust/persia-core/src/forward.rs:71
↓ 1 callersMethodget_type_name
Name of current datatype
rust/persia-core/src/tensor.rs:65
↓ 1 callersMethodget_type_size
Bit size of current datatype.
rust/persia-core/src/tensor.rs:47
↓ 1 callersMethodhashmap_key
(&self)
rust/persia-embedding-holder/src/emb_entry.rs:161
↓ 1 callersMethodinc_dir_scan_thread
(&self)
rust/persia-incremental-update-manager/src/lib.rs:314
↓ 1 callersFunctionindices_to_sharded_indices
( indices: &IDTypeFeatureBatch, config: &EmbeddingConfig, replica_size: u64, )
rust/persia-embedding-server/src/embedding_worker_service/mod.rs:454
↓ 1 callersFunctioninfer
(stub, model_name, model_input)
examples/src/adult-income/serve_client.py:26
↓ 1 callersMethodinit_adagrad
( &mut self, lr: f32, wd: f32, g_square_momentum: f32, initialization:
rust/persia-core/src/optim.rs:26
↓ 1 callersMethodinit_adam
(&mut self, lr: f32, betas: (f32, f32), eps: f32)
rust/persia-core/src/optim.rs:51
↓ 1 callersMethodinit_sgd
(&mut self, lr: f32, wd: f32)
rust/persia-core/src/optim.rs:46
↓ 1 callersMethodinit_with_env_file
Check if the current option was initialized with a ddp env file or not Returns: ``True`` if the current option was initialized wi
persia/distributed.py:195
↓ 1 callersFunctioninitialize_dataflow
(world_size: usize, channel: &PersiaBatchDataSender)
rust/persia-core/src/nats.rs:410
↓ 1 callersMethodis_master_server
(&self)
rust/persia-model-manager/src/lib.rs:115
↓ 1 callersMethodis_master_server
(&self)
rust/persia-embedding-server/src/embedding_worker_service/mod.rs:651
↓ 1 callersMethodis_raw_embedding
(&self)
rust/persia-core/src/forward.rs:64
↓ 1 callersMethodload_embedding
Load embeddings from ``src_dir``. By default, this function is synchronous and will wait for the completion of embedding loading befor
persia/ctx.py:539
↓ 1 callersMethodload_embedding_checkpoint_info
( &self, emb_dir: &PathBuf, )
rust/persia-model-manager/src/lib.rs:187
↓ 1 callersMethodload_embedding_from_dir
( &self, dir: PathBuf, embedding_holder: PersiaEmbeddingHolder, )
rust/persia-model-manager/src/lib.rs:375
↓ 1 callersMethodload_embedding_from_file
(&self, file_path: PathBuf)
rust/persia-incremental-update-manager/src/lib.rs:224
↓ 1 callersMethodload_embedding_via_emb_servers
( &self, req: String, )
rust/persia-embedding-server/src/embedding_worker_service/mod.rs:1166
↓ 1 callersMethodload_embedding_via_embedding_worker
( &self, req: String, num_model_shards: usize, )
rust/persia-embedding-server/src/embedding_worker_service/mod.rs:1192
↓ 1 callersMethodload_internal_shard_embeddings
( &self, file_path: PathBuf, embedding_holder: PersiaEmbeddingHolder, )
rust/persia-model-manager/src/lib.rs:260
← previousnext →201–300 of 720, ranked by callers