functorch
functorch copied to clipboard
wrapper->level().value() <= current_level INTERNAL ASSERT FAILED at "../aten/src/ATen/functorch/ADInterpreters.cpp":39
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!
Do you have a script we could use to reproduce the problem?