oneAPI-samples icon indicating copy to clipboard operation
oneAPI-samples copied to clipboard

Is multi-gpu supported in oneapi?

Open kakulo opened this issue 3 years ago • 1 comments

I am running a few CNN/DNN examples within oneDNN, but it always run on a single Nvidia GPU. Is there a way to run on multiple Nivida GPUs? Is it currently supported or do I miss something? Thanks!

kakulo avatar May 12 '22 23:05 kakulo

oneDNN supports computation & memory primitives. For communication support such as all reduce among multiple devices, oneCCL supports those communications operations. However, you might need to use NCCL for those communication operations on NVidia GPUs.

aice-support avatar Jul 27 '22 07:07 aice-support

oneDNN supports computation & memory primitives. For communication support such as all reduce among multiple devices, oneCCL supports those communications operations. However, you might need to use NCCL for those communication operations on NVidia GPUs.

@kakulo Was that able to answer your question?

yehudaorel avatar Nov 01 '22 01:11 yehudaorel

@kakulo Hope our response solves your problem. We will close the issue for now, and feel free to reopen the issue if you still have problem.

aice-support avatar Nov 16 '22 18:11 aice-support