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

[ElegantLin]

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!

[zou3519]

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