From ce9894c8bfda0e999d5fdcec33a96ba098976e43 Mon Sep 17 00:00:00 2001
From: Matthew Flatt <mflatt@racket-lang.org>
Date: Wed, 25 Oct 2017 09:55:41 -0700
Subject: [PATCH] JIT: fix inline allocation for large vectors

Cloaes #1800
---
 pkgs/racket-test-core/tests/racket/jitinline.rktl | 12 ++++++++++++
 racket/src/racket/src/jitinline.c                 |  5 ++++-
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/pkgs/racket-test-core/tests/racket/jitinline.rktl b/pkgs/racket-test-core/tests/racket/jitinline.rktl
index c0d8f967e5..ed450aa203 100644
--- a/pkgs/racket-test-core/tests/racket/jitinline.rktl
+++ b/pkgs/racket-test-core/tests/racket/jitinline.rktl
@@ -914,6 +914,18 @@
 
     ))
 
+;; ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
+;; Make sure that the JIT doesn't try to inline
+;; a vector allocation that is too large
+
+(parameterize ([current-namespace (make-base-namespace)])
+  (let loop ([i 10])
+    ((eval `(lambda (x) (vector x ,@(for/list ([j (in-range i)])
+                                      j))))
+     i)
+    (when (i . < . 100000)
+      (loop (floor (* i #e1.25))))))
+
 ;; ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 
 ;; Check JIT handling of structure-reference sequences
diff --git a/racket/src/racket/src/jitinline.c b/racket/src/racket/src/jitinline.c
index da8899e112..d1d8ccc3ef 100644
--- a/racket/src/racket/src/jitinline.c
+++ b/racket/src/racket/src/jitinline.c
@@ -5250,7 +5250,10 @@ static int generate_vector_alloc(mz_jit_state *jitter, Scheme_Object *rator,
     c = 2;
   } else {
     c = app->num_args;
-    if (c)
+    if (c > 256) {
+      /* Too big for inline alloc */
+      return scheme_generate_app(app, NULL, c, c, jitter, 0, 0, 0, 0);
+    } else if (c)
       scheme_generate_app(app, NULL, c, c, jitter, 0, 0, 0, 2);  /* sync'd below */
   }
   CHECK_LIMIT();