pytorch/test/cpp
Wang, Eikan ef0873327e [NNC] Add utility functions to check channels-last contiguous (#75938)
Summary:
The `Buf` uses `std::vector<ExprHandle>` to represent its strides. The `ExprHandle` could be an immediate value or a mathematical expression with variables involved both for the static shape and dynamic shape. So it is hard to directly deduce the channels-last contiguous layout based on the numerical calculation. Hence, the utility functions of this PR are based on the pattern match to check whether the `Buf` is channels-last contiguous.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/75938

Reviewed By: cpuhrsch

Differential Revision: D35724091

Pulled By: ZolotukhinM

fbshipit-source-id: f79ae21749d0aad8601f0434b52df88602ff09bf
(cherry picked from commit 3712bbbe4bea57c5c1abe1eafde4b8778e13e0c4)
2022-04-22 06:42:39 -07:00
..
api Extend jit::load to work on flatbuffer file; Take 2 (#75256) 2022-04-06 17:54:01 +00:00
c10d Add cuda-11.3+clang9 build workflow (take 2) 2022-04-11 17:13:01 +00:00
common
dist_autograd
jit List of SymInt rebase on master 2022-04-20 02:09:55 +00:00
lazy Decouple Lazy Node Shape Cache (#75324) 2022-04-21 17:27:05 -07:00
lite_interpreter_runtime [PyTorch] Add codegen unboxing ability (#69881) 2022-03-01 23:28:13 +00:00
monitor
profiler [RecordFunction] More effecient machinery to determine which callbacks to run. (#75807) 2022-04-19 20:46:16 +00:00
rpc
tensorexpr [NNC] Add utility functions to check channels-last contiguous (#75938) 2022-04-22 06:42:39 -07:00
__init__.py