Functions — github.com/aws/aws-ofi-nccl

CodeHub MCPcopy Create free account

hub / github.com/aws/aws-ofi-nccl / functions

Functions1,193 in github.com/aws/aws-ofi-nccl

⨍ Functions 1,193◇ Types & classes 476

↓ 3 callersFunctioninsert_addresses

tests/unit/ep_addr_list.cpp:13

↓ 3 callersMethodinsert_entry

src/nccl_ofi_mr.cpp:100

↓ 3 callersFunctionis_gdr_supported_nic

tests/functional/functional_test.cpp:76

↓ 3 callersMethodlookup_entry

src/nccl_ofi_mr.cpp:61

↓ 3 callersMethodmap

* Map a BAR region of `size` bytes starting at `bar` for GPU * access. After this succeeds, `host` holds the BAR pointer and * `dev` holds the GPU

include/rdma/gin/nccl_ofi_gin_gdaki_resources.h:135

↓ 3 callersFunctionnccl_net_ofi_alloc_mr_buffer

* @brief Allocate memory region for memory registration * * This function allocates memory that covers full page aligned. * * Internally allocated

src/nccl_ofi_net.cpp:114

↓ 3 callersFunctionnccl_net_ofi_devices

src/nccl_ofi_api.cpp:111

↓ 3 callersFunctionnccl_net_ofi_fini

src/nccl_ofi_api.cpp:91

↓ 3 callersFunctionnccl_net_ofi_get_gpu_device_for_addr

src/nccl_ofi_cuda.cpp:279

↓ 3 callersFunctionnccl_net_ofi_get_plugin

src/nccl_ofi_api.cpp:105

↓ 3 callersFunctionnccl_net_ofi_gettid

src/nccl_ofi_compat.cpp:13

↓ 3 callersFunctionnccl_net_ofi_gpu_mem_copy_host_to_device

src/nccl_ofi_cuda.cpp:221

↓ 3 callersFunctionnccl_net_ofi_iflush

src/nccl_ofi_api.cpp:528

↓ 3 callersFunctionnccl_net_ofi_mem_defined

include/nccl_ofi_memcheck_nop.h:15

↓ 3 callersFunctionnccl_ofi_dlist_for_each_safe

tests/unit/ctrl_msg.cpp:187

↓ 3 callersFunctionnccl_ofi_dmabuf_viable

Check preconditions for using DMA-BUF. Note that we may disable DMA-BUF for * other reasons, even if this function returns true. For example, if we d

src/nccl_ofi_dmabuf.cpp:51

↓ 3 callersFunctionnccl_ofi_mr_ckey_fill_mr_attrs

include/nccl_ofi_mr.h:160

↓ 3 callersFunctionnccl_ofi_mr_ckey_mk_dmabuf

include/nccl_ofi_mr.h:121

↓ 3 callersFunctionnccl_ofi_mr_ckey_type_str

include/nccl_ofi_mr.h:58

↓ 3 callersFunctionnccl_ofi_ofiutils_cq_create

* @brief Create and initialize libfabric completion queue */

src/nccl_ofi_ofiutils.cpp:440

↓ 3 callersFunctionnccl_ofi_ofiutils_fabric_create

* @brief Create and initialize libfabric fabric */

src/nccl_ofi_ofiutils.cpp:226

↓ 3 callersFunctionnccl_ofi_topo_free

src/nccl_ofi_topo.cpp:194

↓ 3 callersFunctionofi_info_get_pci_attr

* @brief Return pointer to fi_pci_attr struct of libfabric NIC info * * @return Pointer to fi_pci_attr struct, if available * NULL, on others */

src/nccl_ofi_topo.cpp:128

↓ 3 callersMethodon_list

True if this node is currently linked into a list. */

include/nccl_ofi_dlist.h:40

↓ 3 callersMethodpop_front_if

Remove and return the first node if pred returns true, or nullptr if empty or pred returns false. */

include/nccl_ofi_dlist.h:79

↓ 3 callersMethodpost_rx_buffs_on_rail

src/nccl_ofi_rdma.cpp:2230

↓ 3 callersMethodprocess_pending_reqs

src/cm/nccl_ofi_cm_resources.cpp:104

↓ 3 callersMethodprogress

src/rdma/gin/nccl_ofi_gin_resources.cpp:554

↓ 3 callersFunctionrecv_comm_destroy

src/nccl_ofi_rdma.cpp:3698

↓ 3 callersMethodregister_region

src/nccl_ofi_gdrcopy.cpp:108

↓ 3 callersMethodrelease

* Called when an associated communicator is closed */

include/rdma/gin/nccl_ofi_gin_resources.h:387

↓ 3 callersMethodrelease_rdma_ep_resources

src/nccl_ofi_rdma.cpp:6760

↓ 3 callersFunctionreq_get_send_close_data

* @brief Return send close data struct of send close request */

src/nccl_ofi_rdma.cpp:487

↓ 3 callersMethodsend

src/cm/nccl_ofi_cm_resources.cpp:276

↓ 3 callersFunctionsendrecv_allocate_req

* @brief Assign an allocated sendrecv request buffer */

src/nccl_ofi_sendrecv.cpp:870

↓ 3 callersFunctionsendrecv_req_get_ofi_context

src/nccl_ofi_sendrecv.cpp:183

↓ 3 callersFunctionsendrecv_req_update

* @brief Update nccl_ofi_req on completion * Fill up request context to deliver to user along with state update. * User polls state field to check

src/nccl_ofi_sendrecv.cpp:132

↓ 3 callersMethodtest

src/rdma/gin/nccl_ofi_gin_reqs.cpp:199

↓ 3 callersMethodtransform_pow2

include/tuner/nccl_ofi_tuner_region.h:103

↓ 2 callersMethodadd_req

src/cm/nccl_ofi_cm_resources.cpp:98

↓ 2 callersMethodassign_device

include/nccl_ofi.h:785

↓ 2 callersMethodav_insert_address

src/cm/nccl_ofi_cm_resources.cpp:58

↓ 2 callersMethodbuild

src/rdma/gin/nccl_ofi_gin_gdaki_resources.cpp:138

↓ 2 callersFunctioncomm_close_handler

* To be called on every send or recv comm close. Processes all pending * comms, and blocks for completion on the last comm close * * Note: caller m

src/nccl_ofi_rdma.cpp:4082

↓ 2 callersMethodcompute_page_address

src/nccl_ofi_mr.cpp:52

↓ 2 callersMethoddev

include/rdma/gin/nccl_ofi_gin_gdaki_resources.h:245

↓ 2 callersMethoddrain_recv_eager_queue

* @brief Drain the receiver-side ordered eager queue. * * Processes eager entries in offset order against posted recv requests. * - Single recv: co

src/nccl_ofi_rdma.cpp:956

↓ 2 callersMethoddrain_sender_eager_queue

* @brief Drain the sender eager queue against available ctrl msgs. * * For each ctrl msg at next_msg_seq_num: * - Single recv: pop front of queue

src/nccl_ofi_rdma.cpp:5810

↓ 2 callersMethodfini_rx_buffers

src/nccl_ofi_rdma.cpp:6354

↓ 2 callersMethodforced_pcie_copy

src/nccl_ofi_gdrcopy.cpp:208

↓ 2 callersMethodget_coll_info

(self, coll_type: NCCLFunc, msg_size: int, num_pipe_ops: int = 1)

contrib/python/ofi_nccl/tuner/cli/wrapper.py:147

↓ 2 callersMethodget_conn_msg_data

* Return transport-specific connect message data from sender, after * connection is established * * Note: Returns a pointer to memory owned by t

include/cm/nccl_ofi_cm.h:37

↓ 2 callersMethodget_conn_resp_msg_data

src/cm/nccl_ofi_cm.cpp:259

↓ 2 callersFunctionget_device_property

* @brief Return PCI device property of PCI device * * This function reads first `MAX_DEV_PROPERTY_LENGTH` characters from * device property file *

src/nccl_ofi_topo.cpp:1202

↓ 2 callersMethodget_handle

* Obtain the handle associated with this listener * * The caller (transport) should return this handle back to NCCL (see * the handle argument t

include/cm/nccl_ofi_cm.h:100

↓ 2 callersFunctionget_info_for_node

* @brief Return libfabric NIC info struct from info list that corresponds to input topology node * * @param node * The topology node * @param inf

src/nccl_ofi_topo.cpp:381

↓ 2 callersMethodget_next_connector_id

* Return the next available connector ID. Used for listeners and * send_connectors. */

include/cm/nccl_ofi_cm_resources.h:255

↓ 2 callersMethodget_nranks

include/rdma/gin/nccl_ofi_gin.h:232

↓ 2 callersMethodget_ofi_info

include/nccl_ofi_rdma.h:1750

↓ 2 callersMethodget_ofi_info_for_cm

include/nccl_ofi_rdma.h:1744

↓ 2 callersFunctionget_pci_device_speed

* @brief Read link speed and width of PCI device or bridge from file system * * @param node * PCI device or bridge topology node * @param is_nic

src/nccl_ofi_topo.cpp:1277

↓ 2 callersMethodget_properties

src/nccl_ofi_rdma.cpp:403

↓ 2 callersMethodget_req

src/nccl_ofi_rdma.cpp:1414

↓ 2 callersMethodget_rx_buff_fl

include/rdma/gin/nccl_ofi_gin_resources.h:226

↓ 2 callersMethodget_scheduler

include/rdma/gin/nccl_ofi_gin_resources.h:90

↓ 2 callersFunctiongin_cursor_inc

Helpers for the cursor ring. Cursors are uint32_t in memory but we never let them advance past GIN_RX_CONSUMED_MASK, so subtraction is performed

include/rdma/gin/nccl_ofi_gin.h:144

↓ 2 callersFunctionin_list

src/nccl_ofi_ofiutils.cpp:28

↓ 2 callersFunctioninit_rma_op_req

src/nccl_ofi_rdma.cpp:4373

↓ 2 callersMethodinsert

include/stats/histogram.h:36

↓ 2 callersFunctionintersect

* @brief Check if the x0->x1 segment crosses the y0->y1 segment. * * @param x0 * point at the beginning of the first segment. * @param x1 * poi

src/tuner/nccl_ofi_regions.cpp:76

↓ 2 callersMethodinvalidate

src/nccl_ofi_net.cpp:922

↓ 2 callersMethodiputSignal

src/rdma/gin/nccl_ofi_gin.cpp:449

↓ 2 callersFunctionis_routable_interface

src/nccl_ofi_system.cpp:28

↓ 2 callersFunctionnccl_net_ofi_closeListen

src/nccl_ofi_api.cpp:622

↓ 2 callersFunctionnccl_net_ofi_closeRecv

* @brief Destroy receive communicator */

src/nccl_ofi_api.cpp:598

↓ 2 callersFunctionnccl_net_ofi_closeSend

* @brief Destroy send communicator */

src/nccl_ofi_api.cpp:571

↓ 2 callersFunctionnccl_net_ofi_configure_nccl_proto_simple

src/nccl_ofi_net.cpp:963

↓ 2 callersFunctionnccl_net_ofi_gpu_flush_gpudirect_rdma_writes

src/nccl_ofi_cuda.cpp:186

↓ 2 callersFunctionnccl_net_ofi_gpu_host_unregister

src/nccl_ofi_cuda.cpp:234

↓ 2 callersFunctionnccl_net_ofi_gpu_init

src/nccl_ofi_cuda.cpp:92

↓ 2 callersMethodnccl_net_ofi_info_properties

src/nccl_ofi_net.cpp:504

↓ 2 callersFunctionnccl_net_ofi_mem_noaccess

include/nccl_ofi_memcheck_asan.h:30

↓ 2 callersFunctionnccl_net_ofi_query_provider_capabilities

src/nccl_ofi_net.cpp:641

↓ 2 callersFunctionnccl_net_ofi_rdma_init

src/nccl_ofi_rdma.cpp:7492

↓ 2 callersFunctionnccl_net_ofi_req_str

* @brief Print NCCL OFI request information */

src/nccl_ofi_rdma.cpp:1393

↓ 2 callersFunctionnccl_net_ofi_sendrecv_init

src/nccl_ofi_sendrecv.cpp:2282

↓ 2 callersFunctionnccl_net_ofi_test

src/nccl_ofi_api.cpp:515

↓ 2 callersFunctionnccl_ofi_dmabuf_viable_and_supported

* Consolidate dmabuf viable checks with other firmwware related checks */

src/nccl_ofi_dmabuf.cpp:101

↓ 2 callersFunctionnccl_ofi_get_unique_node_id

src/nccl_ofi_system.cpp:55

↓ 2 callersFunctionnccl_ofi_gin_iputSignal

src/rdma/gin/nccl_ofi_gin_api.cpp:338

↓ 2 callersFunctionnccl_ofi_ofiutils_domain_create

* @brief Create and initialize libfabric domain */

src/nccl_ofi_ofiutils.cpp:245

↓ 2 callersFunctionnccl_ofi_topo_create

src/nccl_ofi_topo.cpp:717

↓ 2 callersFunctionnccl_ofi_tuner_compute_cost

src/tuner/nccl_ofi_model.cpp:46

↓ 2 callersFunctionnccl_ofi_tuner_destroy

src/tuner/nccl_ofi_tuner.cpp:30

↓ 2 callersFunctionnccl_ofi_tuner_init

src/tuner/nccl_ofi_tuner.cpp:47

↓ 2 callersFunctionpost_rx_buffer

src/nccl_ofi_rdma.cpp:5366

↓ 2 callersMethodpost_rx_buffs

src/nccl_ofi_rdma.cpp:2276

↓ 2 callersFunctionprint_dev_props

tests/functional/functional_test.cpp:62

↓ 2 callersMethodprint_stats

include/stats/histogram.h:53

↓ 2 callersFunctionpurge_expired_weak_ptrs

src/nccl_ofi_net.cpp:54

← previous next →201–300 of 1,193, ranked by callers