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.

Sign up for GitHub

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

Jump to bottom

Run the GCE JAX test on nightly, stable and stable stack #468

Open
wants to merge 24 commits into
base: master
Choose a base branch
from
Open

Run the GCE JAX test on nightly, stable and stable stack #468

wants to merge 24 commits into from

Conversation

guptaaka
Copy link
Collaborator

@guptaaka guptaaka commented Nov 8, 2024

Description

Run the Jax.distributed.initialize() test on nightly, stable and stable stack JAX versions

Tests

Tested on the dev branch: http://shortn/_WUwhuAqJfm

Checklist

Before submitting this PR, please make sure (put X in square brackets):

  • I have performed a self-review of my code.
  • I have necessary comments in my code, particularly in hard-to-understand areas.
  • I have run one-shot tests and provided workload links above if applicable.
  • I have made or will make corresponding changes to the doc if needed.

guptaaka added 24 commits October 25, 2024 01:53
@guptaaka
Attempt to add JAX functional tests
4642cfb
@guptaaka
fix ()
9dcadc8
@guptaaka
Fix the setup command
08e6f6c
@guptaaka
Fixes
0d4a54c
@guptaaka
Schedule the test
94eb764
@guptaaka
Delete dags/multipod/configs/jax_tests_gke_config.py
aa1787c
@guptaaka
Update cloud_composer.auto.tfvars
c20a099
@guptaaka
Delete deployment/deployment_composer_env/.terraform.lock.hcl
45b8bcf
@guptaaka
Update main.tf
411ab31
@guptaaka
Update cloud_composer.auto.tfvars
1786a74
@guptaaka
Update provider.tf
41dfea2
@guptaaka
Update cloud_composer.auto.tfvars
622fb0d
@guptaaka
Update cloud_composer.auto.tfvars
821a06e
@guptaaka
Update jax_tests_gce_config.py
edf12d0
@guptaaka
Update jax_functional_tests.py
973f627
@guptaaka
Update jax_functional_tests.py
b33e633
@guptaaka
Update cloud_composer.auto.tfvars
52e6ea2
@guptaaka
Update cloud_composer.auto.tfvars
3734953
@guptaaka
Update cloud_composer.auto.tfvars
6654eba
@guptaaka
Update cloud_composer.auto.tfvars
5342b70
@guptaaka
Update jax_tests_gce_config.py
ca0df82
@guptaaka
PR comments
b084ec1
@guptaaka
Run the GCE JAX test on nightly, stable and stable stack
93d890c
@guptaaka
Run the GCE JAX test on nightly, stable and stable stack
6ffbad3
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@jonb377 jonb377 Awaiting requested review from jonb377 jonb377 is a code owner

@tonyjohnchen tonyjohnchen Awaiting requested review from tonyjohnchen tonyjohnchen is a code owner

@raymondzouu raymondzouu Awaiting requested review from raymondzouu raymondzouu is a code owner

@RissyRan RissyRan Awaiting requested review from RissyRan RissyRan is a code owner

@allenwang28 allenwang28 Awaiting requested review from allenwang28 allenwang28 is a code owner

At least 1 approving review is required to merge this pull request.

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
1 participant
@guptaaka