pytorch/test/cpp
goldenxuett fc3555ce4d [JIT] Modify is_nondeterministic to utilize tags in schemaInfo and integrate with ir.cpp (#81836)
- Modified is_nondeterministic method in SchemaInfo class to utilize tags.
- Modified isNonDeterministic method in ir.cpp to utilize SchemaInfo when a Node is an aten op.
- Added an assert to ensure that if a node is an aten op kind, it has a schema.
- Tested through verifying that all IR.cpp tests run, and through adding 2 custom determinism checks to test for the special dropout edge case and a general bernoulli case.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/81836
Approved by: https://github.com/davidberard98
2022-07-25 15:44:31 +00:00
..
api Lower randint default dtype to the C++ API (#81410) 2022-07-21 16:42:49 +00:00
c10d Build example_allreduce only for GLOO (#81062) 2022-07-08 02:25:54 +00:00
common
dist_autograd [lint] autoformat test/cpp and torch/csrc 2022-06-11 21:11:16 +00:00
jit [JIT] Modify is_nondeterministic to utilize tags in schemaInfo and integrate with ir.cpp (#81836) 2022-07-25 15:44:31 +00:00
lazy [LT] Add a new backend interface [DUP of the original] (#81662) 2022-07-19 01:15:22 +00:00
lite_interpreter_runtime Back out "[profiling] Adding targets file for test_mobile_profiler" (#80789) 2022-07-05 23:34:15 +00:00
monitor
profiler [lint] autoformat test/cpp and torch/csrc 2022-06-11 21:11:16 +00:00
rpc [lint] autoformat test/cpp and torch/csrc 2022-06-11 21:11:16 +00:00
tensorexpr [BE] Use CamelCase for enum class members (#79772) 2022-06-17 05:53:57 +00:00
__init__.py