pytorch
9ee4f499 - [OpBench] add _consume_op.list for processing input with type of List[Tensor] (#47890)

Commit
4 years ago
[OpBench] add _consume_op.list for processing input with type of List[Tensor] (#47890) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/47890 As titled. In order to fix issue when running `chunk_test`, `split_test`, `qobserver` , `sort in qunary` in jit mode, because the output of `chunk_op` is a list of tensors which can not be handled by the current `_consume_op` Test Plan: OSS: python3 -m benchmark_all_test --iterations 1 --warmup_iterations 1 --use_jit Reviewed By: mingzhe09088 Differential Revision: D24774105 fbshipit-source-id: 210a0345b8526ebf3c24f4d0794e20b2ff6cef3d
Author
Yang Wang
Parents
Loading