device_nccl_reduce.c File Reference

#include <stdlib.h>
#include <stdio.h>
#include <comm/comm_nccl.h>
#include "device_nccl_op.h"

Include dependency graph for device_nccl_reduce.c:

Go to the source code of this file.

Functions
void	device_nccl_allreduce (void *sbuf_d, void *rbuf_d, int count, int nbytes, int op, void *stream)

Function Documentation

device_nccl_allreduce()

void device_nccl_allreduce	(	void *	sbuf_d,
		void *	rbuf_d,
		int	count,
		int	nbytes,
		int	op,
		void *	stream
	)

C wrapper for NCCL reduction calls

Definition at line 45 of file device_nccl_reduce.c.

Here is the call graph for this function:

Here is the caller graph for this function: