mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
[ca] skip on some PYTORCH_TEST_WITH_DYNAMO=1 autograd tests (#156374)
These aren't supported. Not sure how they passed CI Pull Request resolved: https://github.com/pytorch/pytorch/pull/156374 Approved by: https://github.com/jansel
This commit is contained in:
committed by
PyTorch MergeBot
parent
fab85fc5f9
commit
f1968a5e76
@ -1580,6 +1580,10 @@ TEST_WITH_TORCHDYNAMO: bool = TestEnvironment.def_flag(
|
||||
env_var="PYTORCH_TEST_WITH_DYNAMO",
|
||||
implied_by_fn=lambda: TEST_WITH_TORCHINDUCTOR or TEST_WITH_AOT_EAGER,
|
||||
)
|
||||
TEST_WITHOUT_COMPILED_AUTOGRAD: bool = TestEnvironment.def_flag(
|
||||
"TEST_WITHOUT_COMPILED_AUTOGRAD",
|
||||
env_var="PYTORCH_TEST_WITHOUT_COMPILED_AUTOGRAD",
|
||||
)
|
||||
|
||||
if TEST_WITH_TORCHDYNAMO:
|
||||
import torch._dynamo
|
||||
@ -1594,7 +1598,7 @@ if TEST_WITH_TORCHDYNAMO:
|
||||
torch._inductor.config.fallback_random = True
|
||||
else:
|
||||
# only dynamo for now
|
||||
torch._dynamo.config.compiled_autograd = True
|
||||
torch._dynamo.config.compiled_autograd = not TEST_WITHOUT_COMPILED_AUTOGRAD
|
||||
|
||||
|
||||
# seems like this is only used in test/torch_np
|
||||
|
Reference in New Issue
Block a user