Hi,
Is it possible to know which chunk of data will be sent to which device?
I have tried get_device()
to know the gpu ID, but I am not sure it is which chunk of data: first split, or second split.
Hi,
Is it possible to know which chunk of data will be sent to which device?
I have tried get_device()
to know the gpu ID, but I am not sure it is which chunk of data: first split, or second split.