pytorch
3dd0e362 - [TensorExpr] Fix min and max for integral inputs in CUDA backend (#44984)

Commit
4 years ago
[TensorExpr] Fix min and max for integral inputs in CUDA backend (#44984) Summary: For integral types, isnan is meaningless. Provide specializations for maximum and minimum which don't call it. Pull Request resolved: https://github.com/pytorch/pytorch/pull/44984 Test Plan: python test/test_jit_fuser_te.py -k TestTEFuser.test_minmax_int_ops Reviewed By: ezyang Differential Revision: D23885259 Pulled By: asuhan fbshipit-source-id: 2e6da2c43c0ed18f0b648a2383d510894c574437
Author
Parents
Loading