Skip to content

Simplify the Bazel logic used to add dependencies to tests. #28753

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

Merged
merged 1 commit into from
May 14, 2025

Conversation

hawkinsp
Copy link
Collaborator

A lot of this logic was confusing phrased as conditions over both CPU and GPU build flags. But we can decompose it:

  • dependencies we add for CPU tests, and
  • additional dependencies we add for GPU tests.

While we are here, also add the necessary pypi dependency for TPU tests.

A lot of this logic was confusing phrased as conditions over both CPU
and GPU build flags. But we can decompose it:
* dependencies we add for CPU tests, and
* additional dependencies we add for GPU tests.

While we are here, also add the necessary pypi dependency for TPU tests.
@hawkinsp hawkinsp requested a review from danielsuo May 14, 2025 20:43
@hawkinsp hawkinsp added the pull ready Ready for copybara import and testing label May 14, 2025
@copybara-service copybara-service bot merged commit 7e8fa0d into jax-ml:main May 14, 2025
25 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
pull ready Ready for copybara import and testing
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant