search

pytorch-labs / float8_experimental

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

[6/x] switch inference tests to use Float8Linear #297

Closed vkuzo closed 3 months ago

vkuzo commented 3 months ago

Stack from ghstack (oldest at bottom):

Summary:

Since inference logic isn't used yet, saving some time and just switching the tests directly instead of testing both versions.

Test Plan:

pytest test/test_inference_flows.py

Reviewers:

Subscribers:

Tasks:

Tags:

Differential Revision: D59305794

vkuzo commented 3 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 3 months ago

This pull request has been merged in pytorch-labs/float8_experimental@88d94072c5541b230087601bd6070b0425598dfe.