unify linear test cases

pytorch-labs / float8_experimental

This repository contains the experimental PyTorch native float8 training UX

BSD 3-Clause "New" or "Revised" License

194 stars 18 forks source link

Closed vkuzo closed 1 week ago

vkuzo commented 1 week ago

Stack from ghstack (oldest at bottom):

Summary:

cleaning up some light tech debt

Test Plan:

pytest test/test_base.py -x -s -k test_linear

Reviewers:

Subscribers:

Tasks:

Tags:

Differential Revision: D59521200

vkuzo commented 1 week ago

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

facebook-github-bot commented 1 week ago

This pull request has been merged in pytorch-labs/float8_experimental@13d319858d8c2e9e0bef0e509ac753b84ac08fa6.