MCPcopy Create free account

hub / github.com/ai-dynamo/nixl / functions

Functions3,108 in github.com/ai-dynamo/nixl

↓ 1 callersMethodget_combine_config
Get a recommended combine config. Argument: num_ranks: the number of ranks. Returns: config: the re
examples/device/ep/nixl_ep/buffer.py:249
↓ 1 callersMethodget_dispatch_config
Get a recommended dispatch config. Argument: num_ranks: the number of ranks. Returns: config: the r
examples/device/ep/nixl_ep/buffer.py:221
↓ 1 callersMethodget_killed_ranks
Get ranks that were killed at current phase.
examples/device/ep/tests/elastic/plan.py:78
↓ 1 callersMethodget_kv_size_per_token
Get the key-value cache size for the GPT-OSS model (per token). The KV cache is based on the Grouped Query Attention mechanism:
benchmark/kvbench/models/gpt_oss.py:63
↓ 1 callersMethodget_kv_size_per_token
Get the key-value cache size for the Llama 3.1 model (per token). Returns: int: The size of the key-value cache, current
benchmark/kvbench/models/llama3_1.py:62
↓ 1 callersMethodget_kv_size_per_token
Get the key-value cache size for the DeepSeek-R1 model. Returns: int: The size of the key-value cache, currently hardcod
benchmark/kvbench/models/deepseek_r1.py:69
↓ 1 callersFunctionget_lib_path
(nixl_root_path: &str, arch: &str)
src/bindings/rust/build.rs:20
↓ 1 callersMethodget_local_ipc_handle
examples/device/ep/csrc/nixl_ep.cpp:264
↓ 1 callersMethodget_low_latency_buffer_size_hint
Get a minimum buffer size requirement for low-latency mode. The size calculation will be done with BF16. Arguments: num_
examples/device/ep/nixl_ep/buffer.py:139
↓ 1 callersMethodget_max_rank
Get the maximum participating rank index
examples/device/ep/tests/elastic/plan.py:86
↓ 1 callersFunctionget_memory_info
Get system memory information from /proc/meminfo
test/unit/plugins/infinia/infinia_nixl_test.cpp:73
↓ 1 callersMethodget_new_ranks
Get ranks to connect to at current phase (only the new ones).
examples/device/ep/tests/elastic/plan.py:38
↓ 1 callersMethodget_next_combine_buffer
examples/device/ep/csrc/nixl_ep.cpp:1245
↓ 1 callersFunctionget_nixl_libs
()
src/bindings/rust/build.rs:68
↓ 1 callersMethodget_notification_bytes
Returns a specific notification for a given agent as raw bytes
src/bindings/rust/src/notify.rs:100
↓ 1 callersMethodget_params
Gets the parameters if this response has a value
src/bindings/rust/src/descriptors/query.rs:108
↓ 1 callersMethodget_rdma_size_hint
Backward-compatible alias for get_low_latency_buffer_size_hint.
examples/device/ep/nixl_ep/buffer.py:162
↓ 1 callersMethodget_removed_ranks
Get ranks to remove at current phase.
examples/device/ep/tests/elastic/plan.py:52
↓ 1 callersFunctionget_repaired_lib_name_map
auditwheel repair renames all libs to include a hash of the library name e.g. "nixl.libs/libboost_atomic-fb1368c6.so.1.66.0" Extract mapp
contrib/wheel_add_ucx_plugins.py:100
↓ 1 callersFunctionget_telemetry_event_type_string
Get string representation of telemetry event type enum
examples/python/telemetry_reader.py:245
↓ 1 callersMethodget_tot_size
src/plugins/gpunetio/verbs/verbs.h:161
↓ 1 callersMethodget_version
Get the buffer version
examples/python/telemetry_reader.py:162
↓ 1 callersMethodhandle
(&self)
src/bindings/rust/src/descriptors/xfer_dlist_handle.rs:29
↓ 1 callersFunctionhandle_remote_transfer_request
Handle remote memory and storage transfers as target.
examples/python/remote_storage_example/nixl_p2p_storage_example.py:229
↓ 1 callersMethodhasExecutor
test/gtest/unit/azure_blob/azure_blob.cpp:150
↓ 1 callersMethodhasExecutor
test/gtest/unit/obj/obj_test_base.h:236
↓ 1 callersMethodhasPcieDevices
@brief Queries whether there is any NIC with PCIe bus info. */
src/utils/libfabric/libfabric_topology.h:242
↓ 1 callersFunctionhas_supported_test_queue
test/unit/plugins/posix/nixl_posix_test.cpp:65
↓ 1 callersFunctionhash_and_size
(file_path)
contrib/wheel_add_ucx_plugins.py:58
↓ 1 callersMethodincrement_completed_requests
src/plugins/libfabric/libfabric_backend.cpp:258
↓ 1 callersFunctioninit_dist
(local_rank: int, num_local_ranks: int)
examples/device/ep/tests/utils.py:34
↓ 1 callersMethodinit_request_tracking
Multi-request completion tracking methods
src/plugins/libfabric/libfabric_backend.cpp:251
↓ 1 callersMethodinitialize
src/plugins/infinia/infinia_client.cpp:50
↓ 1 callersMethodinsertAddress
src/utils/libfabric/libfabric_rail.cpp:1434
↓ 1 callersMethodinsertAllAddresses
src/utils/libfabric/libfabric_rail_manager.cpp:874
↓ 1 callersMethodinto_iter
(self)
src/bindings/rust/src/utils/params.rs:106
↓ 1 callersMethodinvalidate_local_md
Invalidate this agent's metadata in etcd This signals to other agents that this agent's metadata is no longer valid. # Arguments `opt_args` - Option
src/bindings/rust/src/agent.rs:740
↓ 1 callersMethodinvalidate_remote_md
Invalidates a remote metadata for this agent
src/bindings/rust/src/agent.rs:610
↓ 1 callersMethodioDone
src/plugins/posix/posix_backend.cpp:138
↓ 1 callersMethodisComposite
src/plugins/ucx/ucx_backend.cpp:112
↓ 1 callersFunctionisCounterEvent
src/plugins/telemetry/doca/doca_exporter.cpp:69
↓ 1 callersFunctionisEfaDevice
test/unit/utils/libfabric/libfabric_topology_test.cpp:358
↓ 1 callersFunctionisEntireIOto1Bdev
src/plugins/gusli/gusli_backend.cpp:56
↓ 1 callersFunctionisGaugeEvent
src/plugins/telemetry/doca/doca_exporter.cpp:98
↓ 1 callersMethodisLast
test/unit/plugins/mooncake/mooncake_backend_test.cpp:85
↓ 1 callersMethodisLast
test/unit/plugins/ucx/ucx_backend_test.cpp:83
↓ 1 callersMethodisMasterRank
benchmark/nixlbench/src/worker/worker.cpp:183
↓ 1 callersMethodisProperlyInitialized
src/utils/libfabric/libfabric_rail.cpp:628
↓ 1 callersFunctionisValidPrepXferParams
src/plugins/azure_blob/azure_blob_backend.cpp:41
↓ 1 callersFunctionisValidPrepXferParams
src/plugins/posix/posix_backend.cpp:36
↓ 1 callersFunctionisValidPrepXferParams
src/plugins/obj/s3/engine_impl.cpp:20
↓ 1 callersFunctionisValidPrepXferParams
* Validate parameters for prepXfer operation. * Ensures operation type, memory types, and descriptor counts are valid. * * @param operation Transfe
src/plugins/obj/s3_accel/dell/engine_impl.cpp:48
↓ 1 callersMethoditer
Returns an iterator over the parameter key-value pairs
src/bindings/rust/src/utils/params.rs:204
↓ 1 callersMethodloadConnInfo
src/core/nixl_listener.cpp:744
↓ 1 callersMethodloadLocalData
src/infra/nixl_memory_section.cpp:426
↓ 1 callersMethodloadLocalMD
src/plugins/ucx/ucx_backend.cpp:1005
↓ 1 callersFunctionloadPluginList
src/core/nixl_plugin_manager.cpp:201
↓ 1 callersFunctionloadRemote
test/unit/plugins/mooncake/mooncake_backend_test.cpp:338
↓ 1 callersFunctionloadRemote
test/unit/plugins/ucx/ucx_backend_test.cpp:307
↓ 1 callersMethodloadRemoteConnInfo
src/plugins/ucx/ucx_backend.cpp:899
↓ 1 callersMethodloadRemoteData
src/infra/nixl_memory_section.cpp:392
↓ 1 callersMethodloadRemoteMD
Load remote metadata, if supported.
src/api/cpp/backend/backend_engine.h:199
↓ 1 callersMethodloadRemoteSections
src/core/nixl_listener.cpp:780
↓ 1 callersFunctionlogOnPercentStep
Log completion percentage at regular intervals (every log_percent_step percent)
src/plugins/posix/posix_backend.cpp:99
↓ 1 callersMethodlow_latency_combine
A low-latency implementation for combining tokens (reduce **with weights**) with NIXL device API. This kernel requires all the ranks
examples/device/ep/nixl_ep/buffer.py:440
↓ 1 callersMethodlow_latency_dispatch
A low-latency implementation for dispatching with NIXL device API. This kernel requires all the ranks (no matter intranode or interno
examples/device/ep/nixl_ep/buffer.py:328
↓ 1 callersFunctionmain
()
contrib/wheel_add_ucx_plugins.py:269
↓ 1 callersFunctionmain
Args: - prefill_gpus: List of GPUs ranks that are used for prefill - rail_optimized: Whether to reorder the decode workers to mat
benchmark/kvbench/test/inference_workload_matgen.py:337
↓ 1 callersFunctionmain
Main function
examples/python/telemetry_reader.py:261
↓ 1 callersFunctionmain
()
examples/device/ep/tests/elastic/elastic.py:614
↓ 1 callersFunctionmakeHeapAlloc
benchmark/nixlbench/src/worker/nixl/nixl_worker.cpp:465
↓ 1 callersFunctionmakeHugepagesAlloc
benchmark/nixlbench/src/worker/nixl/nixl_worker.cpp:439
↓ 1 callersFunctionmakeMtType
src/plugins/ucx/ucx_utils.cpp:380
↓ 1 callersFunctionmakePublicMetadataRkeys
src/plugins/ucx/ucx_backend.cpp:964
↓ 1 callersFunctionmakeUcpVersion
src/plugins/ucx/ucx_utils.cpp:373
↓ 1 callersFunctionmake_scope_guard
benchmark/nixlbench/src/utils/scope_guard.h:42
↓ 1 callersMethodmatch
src/utils/common/config_traits.h:65
↓ 1 callersFunctionmilliseconds_to_cycles
examples/device/ep/csrc/nixl_ep.cpp:63
↓ 1 callersMethodmin
benchmark/nixlbench/src/utils/utils.cpp:1568
↓ 1 callersFunctionmlx5_init_cqes
src/plugins/gpunetio/verbs/verbs.cpp:64
↓ 1 callersMethodmodifyAlways
src/plugins/ucx/config.cpp:37
↓ 1 callersMethodneedPrep
test/unit/plugins/mooncake/mooncake_backend_test.cpp:70
↓ 1 callersMethodneedPrep
test/unit/plugins/ucx/ucx_backend_test.cpp:70
↓ 1 callersMethodneedRelease
test/unit/plugins/mooncake/mooncake_backend_test.cpp:80
↓ 1 callersMethodneedRelease
test/unit/plugins/ucx/ucx_backend_test.cpp:79
↓ 1 callersFunctionnetif_flags_is_active
src/plugins/gpunetio/gpunetio_utils.cpp:421
↓ 1 callersFunctionnetif_get_addr
src/plugins/gpunetio/gpunetio_utils.cpp:426
↓ 1 callersFunctionneuronFree
benchmark/nixlbench/src/utils/neuron.cpp:176
↓ 1 callersFunctionneuronMalloc
benchmark/nixlbench/src/utils/neuron.cpp:156
↓ 1 callersFunctionneuronMemcpy
benchmark/nixlbench/src/utils/neuron.cpp:184
↓ 1 callersFunctionneuronMemset
benchmark/nixlbench/src/utils/neuron.cpp:198
↓ 1 callersMethodnext
(&mut self)
src/bindings/rust/src/lib.rs:467
↓ 1 callersMethodnext
Advance to next phase.
examples/device/ep/tests/elastic/plan.py:94
↓ 1 callersMethodnixlAgent
It is unsafe to move nixlAgent object */
src/api/cpp/nixl.h:56
↓ 1 callersMethodnixlDocaInitNotif
src/plugins/gpunetio/gpunetio_backend.cpp:398
↓ 1 callersFunctionnixlTelemetryEventTypeForStatus
src/core/telemetry/telemetry.cpp:36
↓ 1 callersFunctionnixlUcxMtLevelIsSupported
src/plugins/ucx/ucx_utils.cpp:354
↓ 1 callersFunctionnixlXferDlistToIOVList
Convert nixl_xfer_dlist_t to vector of xferBenchIOV
benchmark/nixlbench/src/worker/nixl/nixl_worker.cpp:347
↓ 1 callersFunctionnixl_capi_agent_make_connection
src/bindings/rust/stubs.cpp:267
↓ 1 callersFunctionnixl_capi_check_remote_md
src/bindings/rust/stubs.cpp:188
← previousnext →901–1,000 of 3,108, ranked by callers