pytorch
e53b5107 - add addmm op to the benchmark suite (#29783)

Commit
5 years ago
add addmm op to the benchmark suite (#29783) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/29783 Add addmm operator which reuses existing input shapes for the add operator. Test Plan: ``` buck run //caffe2/benchmarks/operator_benchmark/pt:add_test -- --iterations 1 # ---------------------------------------- # PyTorch/Caffe2 Operator Micro-benchmarks # ---------------------------------------- # Tag : short # Benchmarking PyTorch: addmm # Mode: Eager # Name: addmm_M64_N64_K64_cpu # Input: M: 64, N: 64, K: 64, device: cpu Forward Execution Time (us) : 759.237 # Benchmarking PyTorch: addmm # Mode: Eager # Name: addmm_M64_N64_K128_cpu # Input: M: 64, N: 64, K: 128, device: cpu Forward Execution Time (us) : 922.764 # Benchmarking PyTorch: addmm # Mode: Eager # Name: addmm_M64_N64_K64_cpu_bwdall # Input: M: 64, N: 64, K: 64, device: cpu Backward Execution Time (us) : 4689.546 # Benchmarking PyTorch: addmm # Mode: Eager # Name: addmm_M64_N64_K64_cpu_bwd1 # Input: M: 64, N: 64, K: 64, device: cpu Backward Execution Time (us) : 1700.093 # Benchmarking PyTorch: addmm # Mode: Eager # Name: addmm_M64_N64_K64_cpu_bwd2 # Input: M: 64, N: 64, K: 64, device: cpu Backward Execution Time (us) : 2947.427 # Benchmarking PyTorch: addmm # Mode: Eager # Name: addmm_M64_N64_K64_cpu_bwd3 # Input: M: 64, N: 64, K: 64, device: cpu Backward Execution Time (us) : 2518.043 # Benchmarking PyTorch: addmm # Mode: Eager # Name: addmm_M64_N64_K128_cpu_bwdall # Input: M: 64, N: 64, K: 128, device: cpu Backward Execution Time (us) : 5848.369 Reviewed By: bddppq Differential Revision: D18496476 fbshipit-source-id: 4f1c116a2676a64106afa958e8c8a8e109f35a4a
Author
Parents
Loading