pytorch-labs / float8_experimental

This repository contains the experimental PyTorch native float8 training UX
BSD 3-Clause "New" or "Revised" License
211 stars 20 forks source link

delete `swap_linear_with_dynamic` from fsdp2 eager test case #311

Closed vkuzo closed 4 months ago

vkuzo commented 4 months ago

Stack from ghstack (oldest at bottom):

Summary:

clean up some tech debt, the swap_linear_with_dynamic function is redundant with swap_linear_with_float8_linear

Test Plan:

python ./test/test_fsdp2/test_fsdp2_eager.py

Reviewers:

Subscribers:

Tasks:

Tags:

Differential Revision: D59685259

vkuzo commented 4 months ago

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

facebook-github-bot commented 4 months ago

This pull request has been merged in pytorch-labs/float8_experimental@3fe7c4a310a69e7d484b7679e5ba125ce4b0d0cf.