Skip to content

Add Zarr Flag for Pathways#180

Closed
SujeethJinesh wants to merge 1 commit into
mainfrom
sujinesh/add_zarr_pathways_driver
Closed

Add Zarr Flag for Pathways#180
SujeethJinesh wants to merge 1 commit into
mainfrom
sujinesh/add_zarr_pathways_driver

Conversation

@SujeethJinesh
Copy link
Copy Markdown
Contributor

Fixes / Features

  • Use Zarr flag as default for Pathways on Cloud.

Testing / Documentation

Testing details.

  • [ y/n ] Tests pass
  • [ y ] Appropriate changes to documentation are included in the PR

Comment thread src/xpk/core/pathways.py
- --xla_enable_async_all_gather=true
- --pathways_tmp_dir_pattern={args.pathways_gcs_location}"""
- --pathways_tmp_dir_pattern={args.pathways_gcs_location}
- --pathways_experimental_suspend_storage_mode=BACKEND_TENSORSTORE""" # This is to enable zarr for pathways. When OCDBT is tested, remove this flag
Copy link
Copy Markdown

Choose a reason for hiding this comment

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

You do not need to change the PW workers here. Unless you want this flag to be consistent between the 2 jobs, you could remove this flag here.

Comment thread src/xpk/core/pathways.py
- --pathways_tmp_dir_pattern={args.pathways_gcs_location}
- --pathways_expected_instances={expected_instances}"""
- --pathways_expected_instances={expected_instances}
- --pathways_experimental_suspend_storage_mode=BACKEND_TENSORSTORE""" # This is to enable zarr for pathways. When OCDBT is tested, remove this flag
Copy link
Copy Markdown

Choose a reason for hiding this comment

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

Rename "Zarr" to "vanilla TensorStore" in the comment?

@SujeethJinesh
Copy link
Copy Markdown
Contributor Author

Resolved with internal changes instead.

@SujeethJinesh SujeethJinesh deleted the sujinesh/add_zarr_pathways_driver branch August 26, 2024 23:21
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.

2 participants