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

Adds a test comparing the output of torch.compile and export #295

Open drisspg opened 3 months ago

drisspg commented 3 months ago

Summary

Currently blocked by this PR: https://github.com/pytorch/pytorch/pull/129688