EDT + FP16 instability #2549

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

shkarupa-alex opened this issue Aug 25, 2021 · 0 comments · Fixed by #2568

Labels

Contributor

shkarupa-alex commented Aug 25, 2021

System information

OS Platform and Distribution (e.g., Linux Ubuntu 16.04): Colab
TensorFlow version and how it was installed (source or binary): 2.6.0
TensorFlow-Addons version and how it was installed (source or binary): 0.14.0
Python version: 3
Is GPU used? (yes/no): no

Describe the bug
When set output dtype to float16 in some cases result contains Inf values

Code to reproduce the issue
https://colab.research.google.com/drive/1m5W14ZWEPV8eQ1S_M25GyssjYHVulByD?usp=sharing

bhack added the custom-ops label

fsx950223 mentioned this issue

fix edt float16 kernel #2568

Merged

21 tasks

bhack closed this as completed in #2568

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment