flambda-backend: Zero alloc: propagate assume on applications of exte…

…rnals (#2459)

* Add a test

* Propagate assume on applications of primitives

* Fix new test

* Add a test for % builtin

* Regenerate dune.inc

* Fix test failure with heap_allocation

lambda/translcore.ml

@@ -410,8 +410,12 @@ and transl_exp0 ~in_new_scope ~scopes sort e =
         else Rc_normal
       in
       let lam =
+        let loc =
+          map_scopes (update_assume_zero_alloc ~assume_zero_alloc)
+            (of_location ~scopes e.exp_loc)
+        in
         Translprim.transl_primitive_application
-          (of_location ~scopes e.exp_loc) p e.exp_env prim_type
+          loc p e.exp_env prim_type
           ~poly_mode:pmode ~poly_sort:psort
           path prim_exp args (List.map fst arg_exps) position
       in