Update on "Generalize some TensorIterator consumers to take TensorIteratorBase"

ezyang · ezyang · commit 8e534f4b912f · 2020-12-02T10:59:21.000-08:00
Signed-off-by: Edward Z. Yang &lt;ezyang@fb.com&gt;

[ghstack-poisoned]
diff --git a/aten/src/ATen/native/cuda/BinaryAddSubKernel.cu b/aten/src/ATen/native/cuda/BinaryAddSubKernel.cu
@@ -18,7 +18,7 @@ struct AddFunctor {
     scalar_t alpha;
 };
 
-void add_kernel_cuda(TensorIteratorBase& iter, Scalar alpha_scalar) {
+void add_kernel_cuda(TensorIterator& iter, Scalar alpha_scalar) {
   AT_DISPATCH_ALL_TYPES_AND_COMPLEX_AND3(kHalf, kBool, kBFloat16, iter.common_dtype(), "add_cuda/sub_cuda", [&]() {
     AddFunctor<scalar_t> f(alpha_scalar.to<scalar_t>());
     gpu_kernel_with_scalars(iter, f);
