executorch
executorch copied to clipboard
Add GEMM Config to partition Linear, QLinear, DQLinear, Addmm
Summary:
Adding GEMMConfig(XNNPartitionerConfig)
This is a base class which handles grabbing all the dependencies of a gemm like node (conv, addmm, linear) and then validates that the node is a valid partition.
We modify all tests in the linear, and validate that to_edge_transform_and_lower with the new partitioner works with all nodes. Additionally, we add a SingleNodePartitionerConfig
class for simple nodes like q_per_tensor and dq_per_tensor. This helps partition q_per_tensor/dq_per_tensor nodes which are actually explicilty converting the dtype of the input.
Differential Revision: D58903231