You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Investigate why tests/jax/test_ranks.py fails for 0-dim parameters.
This issue happened to me multiple times in some other tests, but I managed to circumvent it. It fails with this error which indicates difference between XLA layout and user input layout (not sure what that is exactly). The error is raised in venv/lib/python3.10/site-packages/jax/_src/interpreters/pxla.py::_get_layouts_from_executable when iterating over input layouts.
Not sure if related, but in BufferInstance::GetMemoryLayout we have some calculations with minor_to_major which previous error indicates is the problem. Take note that when debugging, I couldn't catch GetMemoryLayout in debug log, meaning it wasn't executed at all.
The text was updated successfully, but these errors were encountered:
Investigate why
tests/jax/test_ranks.pyfails for 0-dim parameters.This issue happened to me multiple times in some other tests, but I managed to circumvent it. It fails with this error which indicates difference between XLA layout and user input layout (not sure what that is exactly). The error is raised in
venv/lib/python3.10/site-packages/jax/_src/interpreters/pxla.py::_get_layouts_from_executablewhen iterating over input layouts.Not sure if related, but in
BufferInstance::GetMemoryLayoutwe have some calculations withminor_to_majorwhich previous error indicates is the problem. Take note that when debugging, I couldn't catchGetMemoryLayoutin debug log, meaning it wasn't executed at all.The text was updated successfully, but these errors were encountered: