pytorch/tools/codegen
Jacob Szwejbka 1faba1e4cc [Pytorch Edge] Make RegisterBackendSelect Selective (#59096)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/59096

RegisterBackendSelect is bringing in ~100 extra ops to the runtime. This messes with the compatibility api, and also adds a nontrivial amount of size.

Test Plan: Model Unittests/CI

Reviewed By: iseeyuan

Differential Revision: D28588100

fbshipit-source-id: ffd0b5b9cbe20f27dbf3be418a6c1f80c7396fdb
2021-06-07 19:48:46 -07:00
..
api Change native functions to take c10::string_view args instead of std::string (#57680) 2021-05-20 18:15:45 -07:00
dest remove xla-specific stuff from codegen (minus CPU fallback) (#58064) 2021-05-26 12:52:13 -07:00
selective_build [PyTorch Edge] Eliminate non-determinism when generating build YAML file (#56539) 2021-04-20 17:26:14 -07:00
__init__.py
code_template.py
context.py [codegen] split out backend-specific information from NativeFunction in the model (#57361) 2021-05-17 12:25:35 -07:00
gen_backend_stubs.py [remove xla from codegen] rename aten_xla_type.h -> DispatchKeyNativeFunctions.h (#58568) 2021-05-26 12:53:19 -07:00
gen.py [Pytorch Edge] Make RegisterBackendSelect Selective (#59096) 2021-06-07 19:48:46 -07:00
local.py [PyTorch] Fix const correctness for resize native functions (#55351) 2021-04-21 14:51:41 -07:00
model.py generate inplace/out kernels for xla (#57510) 2021-05-17 12:25:38 -07:00
utils.py Split anonymous and namespaced definitions in RegisterDispatchKey (#51585) 2021-02-04 09:19:41 -08:00