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.

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-sourced update on 12/19/2024 #63

Closed

Conversation

tsunghsienlee
Copy link
Contributor

@tsunghsienlee tsunghsienlee commented Dec 18, 2024

Summary:

  1. Add HybridShardDistributor (i.e., per-parameter HSDP or HSDP2) implemented by wz337 and hjmshi into DistributedShampoo.
  2. Disable quantization functionality for now; as a result, deprecate precision_config.
  3. Deprecate use_pytorch_compile, this is superseded by shampoo_pt2_compile_config.
  4. Deprecate use_protected_eigh, this is always True from now.

Differential Revision: D67398314

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Dec 18, 2024
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D67398314

@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D67398314

tsunghsienlee added a commit to tsunghsienlee/optimizers that referenced this pull request Dec 18, 2024
Summary:
Pull Request resolved: facebookresearch#63

1. Add `HybridShardDistributor` (i.e., per-parameter HSDP or HSDP2) implemented by wz337 and hjmshi into `DistributedShampoo`.
2. Disable quantization functionality for now.

Differential Revision: D67398314
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D67398314

tsunghsienlee added a commit to tsunghsienlee/optimizers that referenced this pull request Dec 18, 2024
Summary:
Pull Request resolved: facebookresearch#63

1. Add `HybridShardDistributor` (i.e., per-parameter HSDP or HSDP2) implemented by wz337 and hjmshi into `DistributedShampoo`.
2. Disable quantization functionality for now.

Differential Revision: D67398314
tsunghsienlee added a commit to tsunghsienlee/optimizers that referenced this pull request Dec 18, 2024
Summary:
Pull Request resolved: facebookresearch#63

1. Add `HybridShardDistributor` (i.e., per-parameter HSDP or HSDP2) implemented by wz337 and hjmshi into `DistributedShampoo`.
2. Disable quantization functionality for now.

Differential Revision: D67398314
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D67398314

1 similar comment
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D67398314

tsunghsienlee added a commit to tsunghsienlee/optimizers that referenced this pull request Dec 18, 2024
Summary:
Pull Request resolved: facebookresearch#63

1. Add `HybridShardDistributor` (i.e., per-parameter HSDP or HSDP2) implemented by wz337 and hjmshi into `DistributedShampoo`.
2. Disable quantization functionality for now.

Differential Revision: D67398314
@tsunghsienlee tsunghsienlee force-pushed the export-D67398314 branch 2 times, most recently from 70a24d9 to c36c9a2 Compare December 18, 2024 23:31
tsunghsienlee added a commit to tsunghsienlee/optimizers that referenced this pull request Dec 18, 2024
Summary:
Pull Request resolved: facebookresearch#63

1. Add `HybridShardDistributor` (i.e., per-parameter HSDP or HSDP2) implemented by wz337 and hjmshi into `DistributedShampoo`.
2. Disable quantization functionality for now.

Differential Revision: D67398314
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D67398314

1 similar comment
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D67398314

tsunghsienlee added a commit to tsunghsienlee/optimizers that referenced this pull request Dec 18, 2024
Summary:
Pull Request resolved: facebookresearch#63

1. Add `HybridShardDistributor` (i.e., per-parameter HSDP or HSDP2) implemented by wz337 and hjmshi into `DistributedShampoo`.
2. Disable quantization functionality for now.

Differential Revision: D67398314
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D67398314

tsunghsienlee added a commit to tsunghsienlee/optimizers that referenced this pull request Dec 18, 2024
Summary:
Pull Request resolved: facebookresearch#63

1. Add `HybridShardDistributor` (i.e., per-parameter HSDP or HSDP2) implemented by wz337 and hjmshi into `DistributedShampoo`.
2. Disable quantization functionality for now.

Differential Revision: D67398314
tsunghsienlee added a commit to tsunghsienlee/optimizers that referenced this pull request Dec 18, 2024
Summary:
Pull Request resolved: facebookresearch#63

1. Add `HybridShardDistributor` (i.e., per-parameter HSDP or HSDP2) implemented by wz337 and hjmshi into `DistributedShampoo`.
2. Disable quantization functionality for now.

Differential Revision: D67398314
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D67398314

1 similar comment
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D67398314

tsunghsienlee added a commit to tsunghsienlee/optimizers that referenced this pull request Dec 19, 2024
Summary:
Pull Request resolved: facebookresearch#63

1. Add `HybridShardDistributor` (i.e., per-parameter HSDP or HSDP2) implemented by wz337 and hjmshi into `DistributedShampoo`.
2. Disable quantization functionality for now.

Differential Revision: D67398314
tsunghsienlee added a commit to tsunghsienlee/optimizers that referenced this pull request Dec 19, 2024
Summary:
Pull Request resolved: facebookresearch#63

1. Add `HybridShardDistributor` (i.e., per-parameter HSDP or HSDP2) implemented by wz337 and hjmshi into `DistributedShampoo`.
2. Disable quantization functionality for now.

Differential Revision: D67398314
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D67398314

1 similar comment
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D67398314

tsunghsienlee added a commit to tsunghsienlee/optimizers that referenced this pull request Dec 19, 2024
Summary:
Pull Request resolved: facebookresearch#63

1. Add `HybridShardDistributor` (i.e., per-parameter HSDP or HSDP2) implemented by wz337 and hjmshi into `DistributedShampoo`.
2. Disable quantization functionality for now.

Differential Revision: D67398314
tsunghsienlee added a commit to tsunghsienlee/optimizers that referenced this pull request Dec 19, 2024
Summary:
Pull Request resolved: facebookresearch#63

1. Add `HybridShardDistributor` (i.e., per-parameter HSDP or HSDP2) implemented by wz337 and hjmshi into `DistributedShampoo`.
2. Disable quantization functionality for now; as a result, deprecate `precision_config`.
3. Deprecate `use_pytorch_compile`, this is superseded by `shampoo_pt2_compile_config`.
4. Deprecate `use_protected_eigh`, this is always `True` from now.

Reviewed By: chuanhaozhuge

Differential Revision: D67398314
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D67398314

Summary:
Pull Request resolved: facebookresearch#63

1. Add `HybridShardDistributor` (i.e., per-parameter HSDP or HSDP2) implemented by wz337 and hjmshi into `DistributedShampoo`.
2. Disable quantization functionality for now; as a result, deprecate `precision_config`.
3. Deprecate `use_pytorch_compile`, this is superseded by `shampoo_pt2_compile_config`.
4. Deprecate `use_protected_eigh`, this is always `True` from now.

Reviewed By: wz337, chuanhaozhuge

Differential Revision: D67398314
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D67398314

@facebook-github-bot
Copy link
Contributor

This pull request has been merged in ae91e28.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported Merged
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants