-
Notifications
You must be signed in to change notification settings - Fork 0
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
Add _test_simple_multiple_platforms_with_extras
#2
Conversation
# 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`. |
There was a problem hiding this comment.
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
.
66c133d
to
040f77e
Compare
040f77e
to
849456e
Compare
Run with: ```console bazel test //tests/pypi/extension:test_simple_multiple_platforms_with_extras ```
1e8ca06
to
392fece
Compare
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.
This reverts commit c1aafac.
Could you please check |
As requested in bazel-contrib#2797 (comment).