Fix hardsigmoid/hardswish for proper device dispatch. (#36704)

Ailing Zhang · facebook-github-bot · commit adca88e82110 · 2020-04-17T09:42:09.000-07:00
Summary: #36351 make `hardsigmoid_backward` use tensoriterator but that can be done only after proper device dispatch. Pull Request resolved: #36704 Differential Revision: D21068126 Pulled By: ailzhang fbshipit-source-id: 6a6a74216f2b50fa7d15f692cd1583d3d233580a
diff --git a/aten/src/ATen/native/native_functions.yaml b/aten/src/ATen/native/native_functions.yaml
@@ -5687,6 +5687,9 @@
 - func: hardsigmoid_backward(Tensor grad_output, Tensor self) -> Tensor
   use_c10_dispatcher: full
   python_module: nn
+  dispatch:
+    CPU: hardsigmoid_backward
+    CUDA: hardsigmoid_backward
 
 - func: hardtanh.out(Tensor self, Scalar min_val=-1, Scalar max_val=1, *, Tensor(a!) out) -> Tensor(a!)
   python_module: nn
@@ -5738,6 +5741,9 @@
 - func: hardswish_backward(Tensor grad_output, Tensor self) -> Tensor
   use_c10_dispatcher: full
   python_module: nn
+  dispatch:
+    CPU: hardswish_backward
+    CUDA: hardswish_backward
 
 - func: leaky_relu.out(Tensor self, Scalar negative_slope=0.01, *, Tensor(a!) out) -> Tensor(a!)
   python_module: nn
