optimize Array.map

Author: bobzhang

jscomp/js_dump.ml

@@ -320,6 +320,19 @@ let rec pp_function method_
           match b.J.expression_desc with 
           | Var (Id i) -> Ident.same a i 
           | _ -> false) l ls ->
+    let optimize  len p cxt f v =
+      if p then
+        begin           
+          P.string f Js_config.curry;
+          P.string f L.dot;
+          P.string f "__";
+          P.string f (Printf.sprintf "%d" len);
+          P.paren_group f 1 (fun _ -> arguments cxt f [function_])            
+        end              
+      else
+        vident cxt f v
+    in
+    let len = List.length l in (* length *)           
     begin match name with 
       | Some i -> 
         P.string f L.var; 
@@ -328,35 +341,15 @@ let rec pp_function method_
         P.space f ;
         P.string f L.eq;
         P.space f ;
-        if arity = NA then
-          let len = List.length l in (* length *)           
-          begin           
-            P.string f Js_config.curry;
-            P.string f L.dot;
-            P.string f "__";
-            P.string f (Printf.sprintf "%d" len);
-            P.paren_group f 1 (fun _ -> arguments cxt f [function_])            
-          end              
-        else
-          vident cxt f v          
+
+        optimize len (arity = NA && len <= 8) cxt f v 
       | None ->
         if return then 
           begin 
             P.string f L.return ;
             P.space f;
           end;
-        if arity = NA then
-          let len = List.length l in (* length *)           
-          begin           
-            P.string f Js_config.curry;
-            P.string f L.dot;
-            P.string f "__";
-            P.string f (Printf.sprintf "%d" len);
-            P.paren_group f 1 (fun _ -> arguments cxt f [function_])            
-          end              
-
-        else           
-          vident cxt f v 
+        optimize len (arity = NA && len <=8) cxt f v 
     end
   | _, _  -> 
 

jscomp/test/.depend

@@ -75,6 +75,8 @@ basic_module_test.cmx : ../stdlib/set.cmx pr6726.cmx offset.cmx \
     mt_global.cmx mt.cmx basic_module_test.cmi
 bdd.cmj : ../stdlib/array.cmi
 bdd.cmx : ../stdlib/array.cmx
+bench.cmj : ../stdlib/array.cmi
+bench.cmx : ../stdlib/array.cmx
 bigarray_test.cmj : ../stdlib/int32.cmi ../stdlib/complex.cmi \
     ../stdlib/bigarray.cmi
 bigarray_test.cmx : ../stdlib/int32.cmx ../stdlib/complex.cmx \
@@ -163,8 +165,8 @@ digest_test.cmx : ../stdlib/string.cmx ../stdlib/printf.cmx mt.cmx \
     ext_array.cmx ../stdlib/digest.cmx ../stdlib/array.cmx
 div_by_zero_test.cmj : mt.cmi ../stdlib/int64.cmi ../stdlib/int32.cmi
 div_by_zero_test.cmx : mt.cmx ../stdlib/int64.cmx ../stdlib/int32.cmx
-earger_curry_test.cmj :
-earger_curry_test.cmx :
+earger_curry_test.cmj : ../stdlib/array.cmi
+earger_curry_test.cmx : ../stdlib/array.cmx
 empty_obj.cmj :
 empty_obj.cmx :
 epsilon_test.cmj : mt.cmi
@@ -843,6 +845,8 @@ basic_module_test.cmj : ../stdlib/set.cmj pr6726.cmj offset.cmj \
     mt_global.cmj mt.cmj basic_module_test.cmi
 bdd.cmo : ../stdlib/array.cmi
 bdd.cmj : ../stdlib/array.cmj
+bench.cmo : ../stdlib/array.cmi
+bench.cmj : ../stdlib/array.cmj
 bigarray_test.cmo : ../stdlib/int32.cmi ../stdlib/complex.cmi \
     ../stdlib/bigarray.cmi
 bigarray_test.cmj : ../stdlib/int32.cmj ../stdlib/complex.cmj \
@@ -931,8 +935,8 @@ digest_test.cmj : ../stdlib/string.cmj ../stdlib/printf.cmj mt.cmj \
     ext_array.cmj ../stdlib/digest.cmj ../stdlib/array.cmj
 div_by_zero_test.cmo : mt.cmi ../stdlib/int64.cmi ../stdlib/int32.cmi
 div_by_zero_test.cmj : mt.cmj ../stdlib/int64.cmj ../stdlib/int32.cmj
-earger_curry_test.cmo :
-earger_curry_test.cmj :
+earger_curry_test.cmo : ../stdlib/array.cmi
+earger_curry_test.cmj : ../stdlib/array.cmj
 empty_obj.cmo :
 empty_obj.cmj :
 epsilon_test.cmo : mt.cmi

jscomp/test/Makefile

@@ -60,7 +60,7 @@ OTHERS := literals a test_ari test_export2 test_internalOO test_obj_simple_ffi t
 	bs_rest_test infer_type_test fs_test module_as_function\
 	test_case_set test_mutliple string_bound_get_test inline_string_test\
 	ppx_this_obj_test unsafe_obj_external gpr_627_test jsoo_485_test jsoo_400_test	\
-	test_require more_uncurry earger_curry_test poly_type
+	test_require more_uncurry earger_curry_test poly_type bench