pytorch
8df5e61f - [nnc] Do not try to vectorize kernels that use float16 (#55970)

Commit
3 years ago
[nnc] Do not try to vectorize kernels that use float16 (#55970) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55970 LLVM's support for float16 is not great, and we were seeing assertion failures trying to generate code for vectorized uses. I note that clang doesn't even try to vectorize operations involving half: https://gcc.godbolt.org/z/86MW4xr17, so that's a good sign we shouldn't either. Fixes #55905 ghstack-source-id: 126511474 Test Plan: pytest test_jit_fuser_te.py -k test_isnan Reviewed By: asuhan Differential Revision: D27752279 Pulled By: bertmaher fbshipit-source-id: ac115080bf2a4a73d52b396d64a5bce0cf13abfe
Author
Parents
Loading