Skip to content
This repository was archived by the owner on Aug 7, 2024. It is now read-only.

move WeightWithDynamicFloat8CastTensor to fsdp_utils.py #310

Closed
wants to merge 2 commits into from

Conversation

vkuzo
Copy link
Contributor

@vkuzo vkuzo commented Jul 9, 2024

Stack from ghstack (oldest at bottom):

Summary:

Refactor in preparation of adding delayed scaling support to float8
all-gather, no logic change.

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:

Differential Revision: D59685262

Summary:

Refactor in preparation of adding delayed scaling support to float8
all-gather, no logic change.

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:

[ghstack-poisoned]
@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 Jul 9, 2024
@vkuzo vkuzo requested review from drisspg and weifengpy July 9, 2024 20:37
Summary:

Refactor in preparation of adding delayed scaling support to float8
all-gather, no logic change.

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:

[ghstack-poisoned]
@vkuzo
Copy link
Contributor Author

vkuzo commented Jul 12, 2024

@vkuzo has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

@facebook-github-bot
Copy link
Contributor

This pull request has been merged in d44a996.

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

Successfully merging this pull request may close these issues.

4 participants