Skip to content

deprecate is_torch_bf16_available - #17738

Merged
sgugger merged 2 commits into
mainfrom
bf16_cpu
Jun 20, 2022
Merged

deprecate is_torch_bf16_available#17738
sgugger merged 2 commits into
mainfrom
bf16_cpu

Conversation

@stas00

@stas00 stas00 commented Jun 16, 2022

Copy link
Copy Markdown
Contributor

This is a follow up to #17734 where @pacman100 discovered that the IPEX PR made is_torch_bf16_available ambiguous, as it went from gpu-only checks to cpu or gpu which is undefined behavior.

So this PR deprecates this function in favor of the very specific is_torch_bf16_gpu_available and is_torch_bf16_cpu_available that were added in #17734

@sgugger

Comment thread src/transformers/training_args.py
@HuggingFaceDocBuilderDev

HuggingFaceDocBuilderDev commented Jun 16, 2022

Copy link
Copy Markdown

The documentation is not available anymore as the PR was closed or merged.

@stas00
stas00 requested a review from sgugger June 17, 2022 15:06

@sgugger sgugger left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Great, thanks a lot for splitting things up like this!

Comment thread src/transformers/file_utils.py Outdated
Comment thread src/transformers/training_args.py
Comment thread src/transformers/utils/import_utils.py Outdated
@stas00

stas00 commented Jun 18, 2022

Copy link
Copy Markdown
Contributor Author

does it look good now, Sylvain?

@sgugger

sgugger commented Jun 20, 2022

Copy link
Copy Markdown
Collaborator

Yes, all good :-) Thanks again!

@sgugger
sgugger merged commit a2d34b7 into main Jun 20, 2022
@sgugger
sgugger deleted the bf16_cpu branch June 20, 2022 12:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants