pytorch
d5d25fec - [primTorch] Adds any, all, equal, item references (#78072)

Commit
2 years ago
[primTorch] Adds any, all, equal, item references (#78072) This PR adds the item, equal, any, and all references. While doing this I found the following issues: - https://github.com/pytorch/pytorch/issues/78070 - https://github.com/pytorch/pytorch/issues/78071 And I fixed a bug where the `convert_element_type` prim could not convert tensors requiring grad to datatypes that don't require grad. Creating the item reference required adding item as a prim, but per @ngimel's suggestion I removed the prims for any and all and implemented them as references, so this is net negative one prim. Reference OpInfos are added for any and all, but item and equal don't even have regular OpInfos. Pull Request resolved: https://github.com/pytorch/pytorch/pull/78072 Approved by: https://github.com/ngimel
Author
Mike Ruberry
Committer
Parents
Loading