frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-11-02 06:24:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix torch square_ benchmark runtime error (#31221)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/31221

This is fixing the runtime error introduced in https://github.com/pytorch/pytorch/pull/30719 that added torch square_ operator to the benchmark suite.

Test Plan:
```
# ----------------------------------------
# PyTorch/Caffe2 Operator Micro-benchmarks
# ----------------------------------------
# Tag : short

# Benchmarking PyTorch: square_
# Mode: Eager
# Name: square__M512_N512_cpu
# Input: M: 512, N: 512, device: cpu
Forward Execution Time (us) : 66.291

Reviewed By: hl475

Differential Revision: D18987889

fbshipit-source-id: 09c56e3a73aab5ab661aac2b06429063b3a82fac
This commit is contained in:
Mingzhe Li
2019-12-12 18:44:24 -08:00
committed by Facebook Github Bot
parent f30b14dead
commit b7c148013f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
aten/src/ATen/native/native_functions.yaml
View File

@ -2566,7 +2566,7 @@
- func: square_(Tensor(a!) self) -> Tensor(a!)
supports_named_tensor: True
variants: method
variants: function, method
- func: std(Tensor self, bool unbiased=True) -> Tensor
use_c10_dispatcher: full