functorch icon indicating copy to clipboard operation
functorch copied to clipboard

wrapper->level().value() <= current_level INTERNAL ASSERT FAILED at "../aten/src/ATen/functorch/ADInterpreters.cpp":39

Open ElegantLin opened this issue 2 years ago • 1 comments

I met this issue when I use jacrev function to calculate the Jacobian matrix using a VisionTransformer. The exception occurs when I move the VisionTransformer to GPU.

RuntimeError('wrapper->level().value() <= current_level INTERNAL ASSERT FAILED at "../aten/src/ATen/functorch/ADInterpreters.cpp":39, please report a bug to PyTorch. escaped?') special variables: function variables: args: ('wrapper->level().val.... escaped?',) special variables: function variables: 0: 'wrapper->level().value() <= current_level INTERNAL ASSERT FAILED at "../aten/src/ATen/functorch/ADInterpreters.cpp":39, please report a bug to PyTorch. escaped?' len(): 1

Could you please help me with this?

Thanks!

ElegantLin avatar Jun 07 '23 05:06 ElegantLin

Do you have a script we could use to reproduce the problem?

zou3519 avatar Jun 08 '23 21:06 zou3519