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