pytorch
fa6df34d - [ET selective build] add kernel metadata section to selective_build.yaml (#100665)

Commit
1 year ago
[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
Author
Committer
Parents
Loading