allow multi-gpu (#31011)

Commit

1 year ago

allow multi-gpu (#31011) * allow multi-gpu * allow multi-gpu --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>