Handle the new device
parameter in dask and demos.
#9386
Merged
device
parameter in dask and demos.
#9386