Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix: allow JAX tracers in from_arraylib #2559

Closed
wants to merge 2 commits into from

Conversation

agoose77
Copy link
Collaborator

@agoose77 agoose77 commented Jul 2, 2023

This should fix #2556, though I'm not sure why we had that check there. It looks intentional. Let's see what the test suite picks up.

@codecov
Copy link

codecov bot commented Jul 2, 2023

Codecov Report

Merging #2559 (1c02005) into main (002a3af) will increase coverage by 0.00%.
The diff coverage is n/a.

Additional details and impacted files
Impacted Files Coverage Δ
src/awkward/_layout.py 84.04% <ø> (+0.36%) ⬆️

@agoose77 agoose77 temporarily deployed to docs-preview July 2, 2023 11:09 — with GitHub Actions Inactive
@agoose77 agoose77 temporarily deployed to docs-preview July 2, 2023 17:20 — with GitHub Actions Inactive
@agoose77
Copy link
Collaborator Author

agoose77 commented Jul 4, 2023

So, the check that this PR removes was added in #1763

I'm not totally clear on what the right solution is here. I think we probably want to allow tracers to be used as buffers for transient layouts, but ensure that the result is tracer free. As such, this PR needs a bit more time.

@agoose77 agoose77 closed this Jan 3, 2024
@jpivarski jpivarski deleted the agoose77/fix-jax-array-from-arraylib branch February 1, 2024 19:24
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Jax tracer TypeError in multiplication
1 participant