pytorch/test/cpp/lazy
Jiewen Tan dc37090ec5 [LT] Support diagonal op (#75230)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/75230

Op diagonal is a view op which we can't code-gen yet. Therefore, support
it by making hand-written IR construction and lowering.

Test Plan: ./build/bin/test_lazy --gtest_filter=LazyOpsTest.TestDiagonal*

Reviewed By: wconstab

Differential Revision: D35378316

Pulled By: alanwaketan

fbshipit-source-id: 7958d00107aef20ac37aabcf2868346240977530
(cherry picked from commit 84155528fce484627c9688cfd92fd4aeb68219e5)
2022-04-08 19:49:42 +00:00
..
CMakeLists.txt Adding integration of SSA into LazyTensor 2022-04-07 19:49:41 +00:00
test_backend_device.cpp Land remaining parts of Torchscript Lazy Tensor backend (#74111) 2022-03-22 23:14:03 +00:00
test_cache.cpp Move shape and operand definitions to base node (#75223) 2022-04-06 01:43:46 +00:00
test_ir_util.cpp Move shape and operand definitions to base node (#75223) 2022-04-06 01:43:46 +00:00
test_ir.cpp Ansley's rebase of DimensionNode onto master (#75352) 2022-04-08 17:22:56 +00:00
test_lazy_ops_util.cpp Fix sign-compare violations in cpp tests 2022-04-04 23:05:31 +00:00
test_lazy_ops_util.h Add lazy tensor unit tests, disabled (#74309) 2022-03-17 15:31:26 +00:00
test_lazy_ops.cpp [LT] Support diagonal op (#75230) 2022-04-08 19:49:42 +00:00
test_misc.cpp Run lazy tensor codegen in generate_code.py (#73996) 2022-03-17 15:31:26 +00:00
test_permutation_util.cpp [LT] Upstream LazyView and view ops IR Nodes (#69277) 2021-12-04 08:44:54 -08:00
test_shape.cpp [LT] Remove torch::lazy::convertShapes (#71291) 2022-01-14 12:06:39 -08:00
test_symbolic_shape.cpp Adding integration of SSA into LazyTensor 2022-04-07 19:49:41 +00:00
test_tensor_impl.cpp Land remaining parts of Torchscript Lazy Tensor backend (#74111) 2022-03-22 23:14:03 +00:00
test_util.cpp [LTC] Upstream some utilities (#69046) 2021-11-30 02:44:02 -08:00