INTERNAL_ASSERT failed

[Jeff09]

RuntimeError: data.find(layerId) == data.end() INTERNAL ASSERT FAILED at "/__w/functorch/functorch/functorch/csrc/DynamicLayer.cpp":251, please report a bug to PyTorch. '

Hi, I have used vmap(jacrev(jac_fn, argnums=0))(*args) to calculate some jacobians, and the above errors appears. Could you help me out? Thank you so much.

[zou3519]

Could you try the latest PyTorch nightly and let us know if the error goes away? The above code no longer exists in trunk (we refactored it away to fix some problems).

[Jeff09]

Thank you for replying. Does the latest PyTorch nightly is the same as PyTorch 2.0?

[Jeff09]

Hi @zou3519 , I have tried PyTorch 1.13.1, it's still the same error.

RuntimeError: data.find(layerId) == data.end() INTERNAL ASSERT FAILED at "../aten/src/ATen/functorch/DynamicLayer.cpp":256, please report a bug to PyTorch.

I can't using PyTorch nightly cause I'm using the library Theseus, which is also from Facebook, which is only support PyTorch 1.13.1, not support PyTorch nightly now. And the bug detail is follows.

Any advices is welcome. Thank you so much.

[zou3519]

Sorry for the delayed reply, I took a long vacation. Is this still a problem? If so, do you have a script that reproduces the problem?