New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open

Hukongtao wants to merge 2 commits into huggingface:main from Hukongtao:fix-non-blocking

+1 −1

Open

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

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

CircleCI Checks / run_tests succeeded Feb 27, 2025 in 8m 48s

Workflow: run_tests

View CircleCI workflow

examples_torch - Success
tests_hub - Success
tests_tokenization - Success
examples_tensorflow - Success
pipelines_tf - Success
collection_job - Success
tests_exotic_models - Success
tests_non_model - Success
pipelines_torch - Success
tests_tf - Success
tests_torch - Success
tests_processors - Success
tests_generate - Success
tests_onnx - Success
tests_custom_tokenizers - Success

View more details on CircleCI Checks