Support set non_blocking=True when move data from cpu to gpu#36408

Open

Hukongtao wants to merge 2 commits intohuggingface:mainfrom Hukongtao:fix-non-blocking

+1-1

Commits on Feb 26, 2025

set non_blocking=True when move data from cpu to gpu
Hukongtao
committed

Commits on Feb 27, 2025

Support set non_blocking=True when move data from cpu to gpu
Hukongtao
committed