Skip to content

Add _test_simple_multiple_platforms_with_extras #2

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

Open
wants to merge 4 commits into
base: exp/pypi-simplify-freethreaded
Choose a base branch
from

Conversation

hartikainen
Copy link

As requested in bazel-contrib#2797 (comment).

@hartikainen hartikainen requested a review from aignas as a code owner August 3, 2025 15:46
Comment on lines 482 to 484
# NOTE(hartikainen): Perhaps this is part of the problem?
# This should say `jax[cuda12]==0.7.0` for `linux_x86_64` platform and
# `jax==0.7.0` for `linux_arm64` and `osx_aarch64`.
Copy link
Author

@hartikainen hartikainen Aug 3, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

My guess is that this points to the main issue. We should probably have two entries under wheel libraries because one of them evaluate to jax[cuda12] and the other two to just jax.

@aignas aignas force-pushed the exp/pypi-simplify-freethreaded branch 2 times, most recently from 66c133d to 040f77e Compare August 10, 2025 15:15
@aignas aignas force-pushed the exp/pypi-simplify-freethreaded branch from 040f77e to 849456e Compare August 17, 2025 01:55
Run with:
```console
bazel test //tests/pypi/extension:test_simple_multiple_platforms_with_extras
```
@hartikainen hartikainen force-pushed the bug/multiple-platforms-with-extras branch from 1e8ca06 to 392fece Compare August 17, 2025 15:34
These changes make `test_simple_multiple_platforms_with_extras` pass but
I don't have enough understanding of the interdependencies here to say
if this is the right solution.
@hartikainen
Copy link
Author

hartikainen commented Aug 17, 2025

I played around with this a little bit added c1aafac that makes test_simple_multiple_platforms_with_extras pass. But I don't have enough understanding of the whole setup here to say if this is at all the right solution. Edit: nevermind, I think c1aafac made no sense at all so I reverted it.

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.

1 participant