↓ 1 callersMethodnccl_comm_init_rank(
&mut self,
host: ResponseId,
curr_time: i64,
device: i32,
rank: i32,
phantora/phantora/src/simulator.rs:858
↓ 1 callersFunctionnccl_comm_split(
rank: i32,
comm_id: *const ffi::c_char,
color: i32,
key: i32,
rank_out: *mut i32,
nr
phantora/cuda_call/src/capi.rs:878
↓ 1 callersFunctionnccl_recv(
count: usize,
dtype: i32,
peer: i32,
comm_id: *const ffi::c_char,
rank: i32,
device:
phantora/cuda_call/src/capi.rs:1009
↓ 1 callersFunctionnccl_reduce_scatter(
count: usize,
dtype: i32,
op: i32,
comm_id: *const ffi::c_char,
rank: i32,
device: i
phantora/cuda_call/src/capi.rs:965
↓ 1 callersFunctionnccl_send(
count: usize,
dtype: i32,
peer: i32,
comm_id: *const ffi::c_char,
rank: i32,
device:
phantora/cuda_call/src/capi.rs:987
Function_phantora_permute(
tokens, routing_map, probs=None, num_out_tokens=None,
fused=False, drop_and_pad=Fals
tests/phantora_utils.py:498