pytorch/tools/jit
Mengwei Liu 797fa26f60 [PyTorch] Only select root ops in codegen unboxing (#74663)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/74663

In lightweight dispatch, we only need to register root ops. Unlike in the dispatcher world, the transitive closure of the operators doesn't need to go through dispatcher or op registry.

Test Plan: Rely on unit tests

Reviewed By: priyaramani

Differential Revision: D35104401

fbshipit-source-id: 1a2df571880ac3c8625985c01bd89a2bb9566af9
(cherry picked from commit 16207fa18e87908ec5e038a7f60f41893a236749)
2022-03-25 02:52:51 +00:00
..
templates Generate aten_interned_strings.h automatically (#69407) 2022-01-18 08:29:54 -08:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00
gen_unboxing.py [PyTorch] Only select root ops in codegen unboxing (#74663) 2022-03-25 02:52:51 +00:00