Adding a few tags in ML_FunctionBasis wrapper schemes

nibrunie · Nov 18, 2018 · ee07120 · ee07120
1 parent 8281af8
commit ee07120
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/metalibm_core/core/ml_function.py b/metalibm_core/core/ml_function.py
@@ -765,15 +765,17 @@ def generate_c_vector_wrapper(self, vector_size, vec_arg_list, vector_scheme, ve
                 LogicalNot(
                   Likely(
                     VectorElementSelection(
-                      vector_mask, vi, precision = ML_Bool
+                      vector_mask, vi, precision = ML_Bool,
+                      tag="vmask_i",
                     ),
                     None
                   ),
                   precision = ML_Bool
                 ),
                 ReferenceAssign(
                   VectorElementSelection(
-                    vec_res, vi, precision = self.precision
+                    vec_res, vi, precision=self.precision,
+                    tag="vres_i"
                   ),
                   scalar_callback(*vec_elt_arg_tuple)
                 )
@@ -801,6 +803,7 @@ def generate_vector_implementation(self, scalar_scheme, scalar_arg_list,
 
     Log.report(Log.Info, "[SV] optimizing Scalar scheme")
     scalar_scheme = self.optimise_scheme(scalar_scheme)
+    scalar_scheme.set_tag("scalar_scheme")
 
     scalar_callback          = scalar_callback_function.get_function_object()