From c8889f4e109866610bd1981f03deee8f102b5ce6 Mon Sep 17 00:00:00 2001 From: Nikita Shulga Date: Wed, 19 Oct 2022 22:15:28 +0000 Subject: [PATCH] `cuda._is_in_bad_fork`->`_C._cuda_isInBadFork` (#87317) Former is always available, while later is only available if PyTorch compiled with CUDA And if it does, then ``` $ python -c "import torch;print(torch._C._cuda_isInBadFork == torch.cuda._is_in_bad_fork)" True ``` Fixes https://github.com/pytorch/torchdynamo/issues/1709 ( at least the symptom) cc @jansel @lezcano @fdrocha Pull Request resolved: https://github.com/pytorch/pytorch/pull/87317 Approved by: https://github.com/voznesenskym, https://github.com/albanD, https://github.com/soumith, https://github.com/jansel --- torch/_dynamo/variables/tensor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch/_dynamo/variables/tensor.py b/torch/_dynamo/variables/tensor.py index 4a4e5008db2..a8db819cb27 100644 --- a/torch/_dynamo/variables/tensor.py +++ b/torch/_dynamo/variables/tensor.py @@ -332,7 +332,7 @@ class TensorVariable(VariableTracker): ) elif ( proxy.node.target == torch._C._DisableFuncTorch - or proxy.node.target == torch._C._cuda_isInBadFork + or proxy.node.target == torch.cuda._is_in_bad_fork ): from . import UserDefinedObjectVariable