Weird lexer warning for high-res initial condition generation

[GeorgKohl]

When installing exponax 0.1.0 via pip:

conda create -n exponax_test python=3.12
conda activate exponax_test
pip install -U "jax[cuda12]"
pip install exponax

Running the following IC generation code leads to a weird lexer warning.

import exponax as ex
import jax

ic_gen = ex.ic.GaussianRandomField(1)
u_init = ic_gen(382, key=jax.random.PRNGKey(0))

The warning looks like this, and also happens with other dimensions, other seeds, or other IC generators. Weirdly, it only happens at resolutions of 382 and above.

E1030 11:07:37.505142  517404 hlo_lexer.cc:443] Failed to parse int literal: 236056074258838375840

[Ceyron]

Thanks for reporting it.

It's a bizarre behavior. I was able to reproduce it.

I guess it is related to some internal behavior in JAX.

@GeorgKohl: Does it also occur with other ic generators?

[GeorgKohl]

Yes, it also happens with other IC generators

[Ceyron]

Okay, thanks for checking. 👍

Since it does not affect the result of the functions (at least so it seems), I would give it a lower priority and check it in the coming weeks.

[GeorgKohl]

So far I only found this on 382², 383², 384², and on 382³, 383³, 384⁴