pytorch/test/cpp
Mikhail Zolotukhin d60efd8207 [TensorExpr] Fix handling of 0-dim tensors. (#59279)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/59279

There were some issues with how we handle 0-dim cases in lowerings and
also in how we generate reductions in that special case. This PR fixes
those issues and reenables a bunch of tests.

Differential Revision:
D28819780
D28819780

Test Plan: Imported from OSS

Reviewed By: navahgar

Pulled By: ZolotukhinM

fbshipit-source-id: f3feff35a1ce11821ada2f8d04ae9d4be10dc736
2021-06-04 13:58:15 -07:00
..
api Add is_inference to native functions (#58729) 2021-06-04 08:59:11 -07:00
common
dist_autograd
jit Make JIT not assume that the device is CUDA. (#54238) 2021-06-03 22:21:27 -07:00
lite_interpreter_runtime [Pytorch Delegated Backend] Save function name in debug info (#57481) 2021-05-25 13:19:02 -07:00
rpc Fix race condition in TP agent (#58753) 2021-06-04 06:53:42 -07:00
tensorexpr [TensorExpr] Fix handling of 0-dim tensors. (#59279) 2021-06-04 13:58:15 -07:00
__init__.py