Hansong Zhang fa6df34d30 [ET selective build] add kernel metadata section to selective_build.yaml (#100665) ... Summary: For each op, we have a List[List[dtype;dim-order]]: - the inner list contains the `dtype;dim-order` info for each arg if we have a Tensor/TensorList/OptionalTensorList - the outer list contains different occurances of dtype/dim-order combinations for that op in the program Example: ``` et_kernel_metadata: aten::add.out: # A list of different dtype/dim-order combinations used in model - # Each contains the list of args of Tensor dtype and dim order if applicable - FLOAT;0,1 - FLOAT;0,1 - NON_TENSOR_ARG - FLOAT;0,1 - FLOAT;0,1 - - INT;0,1 - INT;0,1 - NON_TENSOR_ARG - INT;0,1 - INT;0,1 aten::mul.out: - - FLOAT;0,1 - FLOAT;0,1 - FLOAT;0,1 - FLOAT;0,1 ``` We don't have the arg name so far; we need to parse the schema (functions.yaml) to get that info. We depend on the order of args from that file. Test Plan: `buck run fbcode//executorch/codegen/tools:test_gen_oplist_real_model` Differential Revision: D45551409 Pull Request resolved: https://github.com/pytorch/pytorch/pull/100665 Approved by: https://github.com/larryliu0820		2023-05-09 21:30:01 +00:00
..
__init__.py
operator.py	[BE] Enable flake8-comprehension rule C417 (#97880)	2023-03-30 14:34:24 +00:00
selector.py	[ET selective build] add kernel metadata section to selective_build.yaml (#100665)	2023-05-09 21:30:01 +00:00