Commits on Jul 29, 2024

1. [wip] add axiswise granularity to Float8Tensor …

   Summary:
   
   This PR adds the axiswise scaling granularity to `Float8Tensor` and
   ensures that basic ops like transpose and `torch._scaled_mm` work as
   expected.
   
   A future PR will add integration with `Float8Linear`.
   
   Test Plan:
   
   TODO
   
   Reviewers:
   
   Subscribers:
   
   Tasks:
   
   Tags:
   
   [ghstack-poisoned]

   

   vkuzo committed Jul 29, 2024
   Configuration menu

   • View commit details
   • Copy full SHA for 6dffce9
   • Browse repository at this point

   Copy the full SHA

   6dffce9 View commit details

   Browse the repository at this point in the history