executorch icon indicating copy to clipboard operation
executorch copied to clipboard

Add GEMM Config to partition Linear, QLinear, DQLinear, Addmm

Open mcr229 opened this issue 7 months ago • 3 comments

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

mcr229 avatar Jul 23 '24 18:07 mcr229