diffrax icon indicating copy to clipboard operation
diffrax copied to clipboard
verified publisher icon patrick-kidger

test: add tracer-leak check

Open nstarman opened this issue 2 months ago • 2 comments

nstarman avatar Nov 13 '25 14:11 nstarman

@patrick-kidger in testing quaxify(diffrax) I'm seeing errors that I think originate from having the tracer checks on. This means that diffrax isn't compatible with JAX_CHECK_TRACER_LEAKS=1 and this PR should demonstrate the issue.

nstarman avatar Nov 13 '25 15:11 nstarman

Hmm, bother.

I think fixing the underlying issue comes under "I'd be happy to take a PR on that". We could then add a test at the same time. (Ideally just a single test, to avoid duplicating out our test suite by too much.)

On the other CI changes here - heads-up that we have a dev branch with this updated.

patrick-kidger avatar Nov 15 '25 06:11 patrick-kidger