Open alexbarghi-nv opened 4 months ago
Currently, the device argument is not respected, with the default device being used instead. This should be changed to move data to the desired device instead, regardless of which device is used to process the graph/samples.
device
Currently, the
device
argument is not respected, with the default device being used instead. This should be changed to move data to the desired device instead, regardless of which device is used to process the graph/samples.