Skip to content

[Mosaic] Use native bf16 ops for tanh, exp and log on TPUv6+. #28531

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

Jump to bottom
Closed
copybara-service wants to merge 0 commits into from
Closed

[Mosaic] Use native bf16 ops for tanh, exp and log on TPUv6+. #28531

copybara-service wants to merge 0 commits into from

Conversation

@copybara-service
Copy link

@copybara-service copybara-service bot commented May 6, 2025

[Mosaic] Use native bf16 ops for tanh, exp and log on TPUv6+.

Replace needs_cast condition during canonicalization with need_elementwise_canonicalization.

@copybara-service copybara-service bot force-pushed the test_755108372 branch 2 times, most recently from 96780f2 to e36183b Compare May 20, 2025 04:37
@copybara-service copybara-service bot closed this May 20, 2025
@copybara-service copybara-service bot deleted the test_755108372 branch May 20, 2025 06:28
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

0 participants