pytorch
8b9bac1f - add operator-range argument to the op bench (#30051)

Commit
6 years ago
add operator-range argument to the op bench (#30051) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/30051 This argument takes hyphen delimited start and end chars to filter operators. If the first character of an operator is in the start and end range, it will be tested. Otherwise skipped. (Note: this ignores all push blocking failures!) Test Plan: ``` buck run mode/opt caffe2/benchmarks/operator_benchmark:benchmark_all_test -- --iterations 1 --operator_range b-c # ---------------------------------------- # PyTorch/Caffe2 Operator Micro-benchmarks # ---------------------------------------- # Tag : short # Benchmarking PyTorch: ceil # Mode: Eager # Name: ceil_M512_N512_cpu # Input: M: 512, N: 512, device: cpu Forward Execution Time (us) : 110.720 # Benchmarking PyTorch: ceil_ # Mode: Eager # Name: ceil__M512_N512_cpu # Input: M: 512, N: 512, device: cpu Forward Execution Time (us) : 51.128 ... buck run mode/opt caffe2/benchmarks/operator_benchmark:benchmark_all_test -- --iterations 1 --operator_range None # ---------------------------------------- # PyTorch/Caffe2 Operator Micro-benchmarks # ---------------------------------------- # Tag : short # Benchmarking PyTorch: abs # Mode: Eager # Name: abs_M512_N512_cpu # Input: M: 512, N: 512, device: cpu Forward Execution Time (us) : 107.113 # Benchmarking PyTorch: abs_ # Mode: Eager # Name: abs__M512_N512_cpu # Input: M: 512, N: 512, device: cpu Forward Execution Time (us) : 54.259 ... Reviewed By: hl475 Differential Revision: D18581910 fbshipit-source-id: b1a1a7ba76f4d6a61c8a1659f15e9c66097654d4
Author
Parents
Loading