Commits on May 28, 2024

1. to_edge_transform_and_lower (#3483) …

   Summary:
   
   This diff introduces the to_edge_transform_and_lower API. The changes introduces are:
   - Adding support to the Parititioner class to register ops that it doesn't want to be composed
   - Changes to _program.py to add the implementation of to_edge_transform_and_lower()
   - Added a basic test case to test that Linear, SDPA & Linear + SDPA are not decomposed when asked and the corresponding backend consumes them.
   
   Reviewed By: kimishpatel, mcr229
   
   Differential Revision: D56401086

   

   tarun292 authored and facebook-github-bot committed May 28, 2024
   Configuration menu

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

   Copy the full SHA

   e83342c View commit details

   Browse the repository at this point in the history