pytorch
0ab03d32 - only run embeddingbag benchmark on cpu (#30106)

Commit
5 years ago
only run embeddingbag benchmark on cpu (#30106) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/30106 as title Test Plan: ``` buck run mode/opt //caffe2/benchmarks/operator_benchmark:benchmark_all_other_test -- --tag_filter all --iterations 1 --device cuda --operators embeddingbag # ---------------------------------------- # PyTorch/Caffe2 Operator Micro-benchmarks # ---------------------------------------- # Tag : all Reviewed By: hl475 Differential Revision: D18598198 fbshipit-source-id: 9b7d103410f1183fdf6776047ea2ef8dba4b7831
Author
Parents
Loading