pytorch
90e3bbf3 - support all with tag_filter to run all shapes (#29864)

Commit
5 years ago
support all with tag_filter to run all shapes (#29864) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/29864 This diff make `all` as a reserved keyword for tag_filter. When `all` is passed from user, it will run all the supported shapes. Test Plan: ``` buck run //caffe2/benchmarks/operator_benchmark/pt:add_test -- --iterations 1 --tag_filter all # ---------------------------------------- # PyTorch/Caffe2 Operator Micro-benchmarks # ---------------------------------------- # Tag : all # Benchmarking PyTorch: add # Mode: Eager # Name: add_M8_N32_K256_cpu # Input: M: 8, N: 32, K: 256, device: cpu Forward Execution Time (us) : 6798.688 ... Reviewed By: hl475 Differential Revision: D18520249 fbshipit-source-id: 4d55af9f46f89b2fe8842e1a00dfa8e5acaf4fa2
Author
Parents
Loading