stable on torch 2.11.x. CI currently pins torch 2.10, where the compiled path
can segfault or drift, and other torch versions are similarly unstable. Skip
the compile-parity test off 2.11 until CI standardizes on a SeZM-compatible
torch, then drop this guard.
Author: @torch-2.11
Line: 46
|
module=r"torch\._functorch\._aot_autograd\.autograd_cache", |
|
) |
|
|
|
# TODO(torch-2.11): SeZM's ``torch.compile`` / AOT-export code paths are only |
|
# stable on torch 2.11.x. CI currently pins torch 2.10, where the compiled path |
|
# can segfault or drift, and other torch versions are similarly unstable. Skip |
|
# the compile-parity test off 2.11 until CI standardizes on a SeZM-compatible |
|
# torch, then drop this guard. |
|
_TORCH_VERSION = parse_version(torch.__version__) |
|
_SKIP_OFF_TORCH_211 = (_TORCH_VERSION.major, _TORCH_VERSION.minor) != (2, 11) |
|
_SKIP_OFF_TORCH_211_REASON = ( |
stable on torch 2.11.x. CI currently pins torch 2.10, where the compiled path
can segfault or drift, and other torch versions are similarly unstable. Skip
the compile-parity test off 2.11 until CI standardizes on a SeZM-compatible
torch, then drop this guard.
Author: @torch-2.11
Line: 46
deepmd-kit/source/tests/pt/model/test_sezm_spin_model.py
Lines 43 to 53 in ccaa551