mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 00:20:18 +01:00
7cfd054075
Summary: When we compute contiguity for a tensor with dynamic shapes we first: 1) Try to compute it without guarding. 2) If all shapes hinted, compute it with potentially adding guards. 3) if any input is not hinted, compute it symbolically. sym_is_contiguous return a SymBool that is then either evaluated or guard_or_false can be called on it to avoid data dependent errors. ex: bool is_contiguous = input.sym_is_contiguous().guard_or_false(__FILE__, __LINE__); is_contiguous_or_false is a helper function that does that. In this PR I only handle default contiguity, will follow up with changes for other formats like channel_last . We use this patter in this PR for several locations to avoid DDEs. Test Plan: contbuild & OSS CI, Rollback Plan: Reviewed By: malfet Differential Revision: D77639021 Pull Request resolved: https://github.com/pytorch/pytorch/pull/157472 Approved by: https://github.com/aorenste |
||
|---|---|---|
| .. | ||
| templates | ||
| __init__.py | ||
| BUILD.bazel | ||
| build.bzl | ||
| context.py | ||
| deprecated.yaml | ||
| derivatives.yaml | ||
| gen_annotated_fn_args.py | ||
| gen_autograd_functions.py | ||
| gen_autograd.py | ||
| gen_inplace_or_view_type.py | ||
| gen_python_functions.py | ||
| gen_trace_type.py | ||
| gen_variable_factories.py | ||
| gen_variable_type.py | ||
| gen_view_funcs.py | ||
| load_derivatives.py | ||
| README.md | ||
If you add a file to this directory, you MUST update
torch/CMakeLists.txt and add the file as a dependency to
the add_custom_command call.