pytorch
b8a1070e - [TensorExpr][CPU] Fix bool -> int casting (#46951)

Commit
4 years ago
[TensorExpr][CPU] Fix bool -> int casting (#46951) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/46951 If e.g. we're casting from torch.int -> torch.bool, previously we would just truncate from int32 -> i8. Since torch.bool has 8 bits but only uses one of them, we need to makes sure that one bit is set. Test Plan: Imported from OSS Reviewed By: ansley Differential Revision: D24805253 Pulled By: eellison fbshipit-source-id: af3aa323f10820d189827eb51037adfa7d80fed9
Author
Elias Ellison
Parents
Loading