Add explicit template variable to map

Author: larryliu0820

kernels/optimized/cpu/op_add.cpp

@@ -67,7 +67,7 @@ Tensor& opt_add_out(
         CTYPE b_val = *b.const_data_ptr<CTYPE>();
 
         using Vec = at::vec::Vectorized<CTYPE>;
-        at::vec::map(
+        at::vec::map<CTYPE>(
             [alpha_val, b_val](Vec x) { return x + Vec(alpha_val * b_val); },
             out.mutable_data_ptr<CTYPE>(),
             a.const_data_ptr<CTYPE>(),
@@ -86,7 +86,7 @@ Tensor& opt_add_out(
           CTYPE b_casted = static_cast<CTYPE>(b_val);
 
           using Vec = at::vec::Vectorized<CTYPE>;
-          at::vec::map(
+          at::vec::map<CTYPE>(
               [alpha_val, b_casted](Vec x) {
                 return x + Vec(alpha_val * b_casted);
               },
@@ -140,7 +140,7 @@ Tensor& opt_add_scalar_out(
           ctx, utils::extract_scalar(alpha, &alpha_val), InvalidArgument, );
 
       using Vec = at::vec::Vectorized<CTYPE>;
-      at::vec::map(
+      at::vec::map<CTYPE>(
           [alpha_val, b_casted](Vec x) {
             return x + Vec(alpha_val * b_casted);
           },

kernels/optimized/cpu/op_div.cpp

@@ -86,14 +86,14 @@ Tensor& opt_div_out(
 
           using Vec = at::vec::Vectorized<CTYPE>;
           if (a.numel() == 1) {
-            at::vec::map(
+            at::vec::map<CTYPE>(
                 [scalar_casted](Vec x) { return Vec(scalar_casted) / x; },
                 out.mutable_data_ptr<CTYPE>(),
                 tensor->const_data_ptr<CTYPE>(),
                 out.numel());
           } else {
             Vec inv_scalar_casted_vec(CTYPE(1) / scalar_casted);
-            at::vec::map(
+            at::vec::map<CTYPE>(
                 [inv_scalar_casted_vec](Vec x) {
                   return x * inv_scalar_casted_vec;
                 },
@@ -193,7 +193,7 @@ Tensor& opt_div_scalar_out(
 
         using Vec = at::vec::Vectorized<CTYPE>;
         Vec inv_b_casted_vec(CTYPE(1) / b_casted);
-        at::vec::map(
+        at::vec::map<CTYPE>(
             [inv_b_casted_vec](Vec x) { return x * inv_b_casted_vec; },
             out.mutable_data_ptr<CTYPE>(),
             a.const_data_ptr<CTYPE>(),

kernels/optimized/cpu/op_exp.cpp

@@ -35,7 +35,8 @@ void exp_data(
     const size_t numel,
     CTYPE_OUT* out_data) {
   using Vec = at::vec::Vectorized<CTYPE_IN>;
-  at::vec::map([](Vec x) { return x.exp(); }, out_data, in_data, numel);
+  at::vec::map<CTYPE_IN>(
+      [](Vec x) { return x.exp(); }, out_data, in_data, numel);
 }
 
 /**

kernels/optimized/cpu/op_le.cpp

@@ -95,7 +95,7 @@ Tensor& opt_le_scalar_out(
         ET_EXTRACT_SCALAR(b, b_val);
         CTYPE b_casted = static_cast<CTYPE>(b_val);
         using Vec = at::vec::Vectorized<CTYPE>;
-        at::vec::map(
+        at::vec::map<CTYPE>(
             [b_casted](Vec x) { return x.le(Vec(b_casted)); },
             out.mutable_data_ptr<CTYPE>(),
             a.const_data_ptr<CTYPE>(),

kernels/optimized/cpu/op_mul.cpp

@@ -55,7 +55,7 @@ Tensor& opt_mul_out(
           CTYPE b_casted = static_cast<CTYPE>(b_val);
 
           using Vec = at::vec::Vectorized<CTYPE>;
-          at::vec::map(
+          at::vec::map<CTYPE>(
               [b_casted](Vec x) { return x * Vec(b_casted); },
               out.mutable_data_ptr<CTYPE>(),
               a.const_data_ptr<CTYPE>(),
@@ -173,7 +173,7 @@ Tensor& opt_mul_scalar_out(
       CTYPE b_casted = utils::scalar_to<CTYPE>(b);
 
       using Vec = at::vec::Vectorized<CTYPE>;
-      at::vec::map(
+      at::vec::map<CTYPE>(
           [b_casted](Vec x) { return x * Vec(b_casted); },
           out.mutable_data_ptr<CTYPE>(),
           a.const_data_ptr<CTYPE>(),

kernels/optimized/cpu/op_sub.cpp

@@ -85,15 +85,15 @@ Tensor& opt_sub_out(
 
           using Vec = at::vec::Vectorized<CTYPE>;
           if (a.numel() == 1) {
-            at::vec::map(
+            at::vec::map<CTYPE>(
                 [alpha_val, scalar_casted](Vec x) {
                   return Vec(scalar_casted) - Vec(alpha_val) * x;
                 },
                 out.mutable_data_ptr<CTYPE>(),
                 tensor->const_data_ptr<CTYPE>(),
                 out.numel());
           } else {
-            at::vec::map(
+            at::vec::map<CTYPE>(
                 [alpha_val, scalar_casted](Vec x) {
                   return x - Vec(alpha_val * scalar_casted);
                 },
@@ -148,7 +148,7 @@ Tensor& opt_sub_scalar_out(
           ctx, utils::extract_scalar(alpha, &alpha_val), InvalidArgument, );
 
       using Vec = at::vec::Vectorized<CTYPE>;
-      at::vec::map(
+      at::vec::map<CTYPE>(
           [alpha_val, b_casted](Vec x) {
             return x - Vec(alpha_val * b_casted);
           },