From 2856c66d707e7f86c7e1df2f715ef49ce6060e86 Mon Sep 17 00:00:00 2001
From: AllanZhengYP <zheallan@amazon.com>
Date: Thu, 9 Dec 2021 22:02:15 +0000
Subject: [PATCH] chore(codegen): set JVM memory size to 4gb

---
 codegen/build.gradle.kts | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/codegen/build.gradle.kts b/codegen/build.gradle.kts
index ccf472b1f0b9..412ae9ed159d 100644
--- a/codegen/build.gradle.kts
+++ b/codegen/build.gradle.kts
@@ -90,6 +90,10 @@ subprojects {
             useJUnitPlatform()
         }
 
+        tasks.withType<JavaExec> {
+            jvmArgs = ['-Xms4g', '-Xmx4g']
+        }
+
         // Apply junit 5 and hamcrest test dependencies to all java projects.
         dependencies {
             testImplementation("org.junit.jupiter:junit-jupiter-api:5.4.0")