pytorch
14c28cae - [JIT] Add determinism checks for ops in SchemaInfo subclass (#81000)

Commit
2 years ago
[JIT] Add determinism checks for ops in SchemaInfo subclass (#81000) - Added is_non_deterministic which returns whether a given op is non-deterministic. Currently this is implemented with a hard-coded list of non-deterministic functions copied from ir.cpp in AliasDB, but this will eventually be implemented by returning with a given schema has the non_deterministic tag. - Tested is_non_deterministic method with a deterministic op and a non deterministic op in test_schema_info.cpp **Note that the case for op "aten::dropout(Tensor input, float p, bool train) -> Tensor" which is deterministic whenever "train=false" is not accounted for in this pr and will be fixed in a later pr. Currently "aten::dropout(Tensor input, float p, bool train) -> Tensor" is always considered nondeterministic.** Pull Request resolved: https://github.com/pytorch/pytorch/pull/81000 Approved by: https://github.com/davidberard98
Author
goldenxuett
Committer
Parents
Loading