You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
i-robot ec4cd6933d !26292 Add GPU operator NeighborExchange 4 years ago
..
nccl_collective_gpu_kernel.cc Add NeighborExchange support for GPU 4 years ago
nccl_collective_gpu_kernel.h Add NeighborExchange support for GPU 4 years ago
nccl_gpu_kernel.h GPU supports p2p nccl interfaces 5 years ago
nccl_p2p_gpu_kernel.cc Add NeighborExchange support for GPU 4 years ago
nccl_p2p_gpu_kernel.h Add NeighborExchange support for GPU 4 years ago
nccl_recv_gpu_kernel.cc Fix code for GPU Pipeline 4 years ago
nccl_recv_gpu_kernel.h Fix code for GPU Pipeline 4 years ago
nccl_send_gpu_kernel.cc GPU supports p2p nccl interfaces 5 years ago
nccl_send_gpu_kernel.h add validation of vector size and non-zero validation of denominator for nn gpu operators. 4 years ago
sync_batch_norm_gpu_kernel.cc first commit 4 years ago
sync_batch_norm_gpu_kernel.h first commit 4 years ago
sync_batch_norm_grad_gpu_kernel.cc first commit 4 years ago
sync_batch_norm_grad_gpu_kernel.h first commit 4 years ago