pytorch
aabc87cd - [NNC] Fix HalfChecker when half present but unused (#48068)

Commit
4 years ago
[NNC] Fix HalfChecker when half present but unused (#48068) Summary: Fixes an internally reported issue in the tensorexpr fuser when using FP16 on Cuda. The HalfChecker analysis to determine if we need to define the Half type searches the IR for expressions that use Half. If one of the parameters is of type Half but it (or any other Half expr) are not used in the IR we'll return a false negative. Fix this by adding the parameter list to the HalfChecker. Pull Request resolved: https://github.com/pytorch/pytorch/pull/48068 Reviewed By: ZolotukhinM Differential Revision: D25009680 Pulled By: nickgg fbshipit-source-id: 24fddef06821f130db3d3f45d6d041c7f34a6ab0
Author
Parents
Loading