pytorch/test/cpp
Elias Ellison 0ecf1add1b Introduce function-local settings for executor, expose in c++ (#74012)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/74012

This allows setting an executor on a function. The first use case is use to decompositions in C++ without additional fusion passes etc which might not work with custom tensors like batched tensors/vmap. A subsequent use case might be taking advantage of invokees of JIT execution which guard on certain properties before invocation (such as complete shapes in AOT autograd, rank in lazy tensor).

Test Plan: Imported from OSS

Reviewed By: gchanan

Differential Revision: D34938124

Pulled By: eellison

fbshipit-source-id: cf7a45416457942b872322cab47d871a8336bdb5
(cherry picked from commit 9c600eb9ad0f2173f003e511268e97584edae36d)
2022-03-29 18:38:52 +00:00
..
api Resolve int[]? arguments to new OptionalIntArrayRef class 2022-03-26 01:45:50 +00:00
c10d
common
dist_autograd
jit Introduce function-local settings for executor, expose in c++ (#74012) 2022-03-29 18:38:52 +00:00
lazy Land remaining parts of Torchscript Lazy Tensor backend (#74111) 2022-03-22 23:14:03 +00:00
lite_interpreter_runtime [PyTorch] Add codegen unboxing ability (#69881) 2022-03-01 23:28:13 +00:00
monitor torch/monitor: merge Interval and FixedCount stats (#72009) 2022-01-30 23:21:59 +00:00
profiler [Profiler] Prefer TSC to wall clock when available (#73855) 2022-03-13 18:29:06 +00:00
rpc
tensorexpr [Quant][core] Merged conv packed params and linear packed params (#73486) 2022-03-11 15:18:45 +00:00
__init__.py