Build: Rework test workers calculation with respect to available memory

#KTI-787

compiler/fir/analysis-tests/build.gradle.kts

@@ -59,9 +59,6 @@ if (kotlinBuildProperties.isInJpsBuildIdeaSync) {
 projectTest(jUnitMode = JUnitMode.JUnit5) {
     dependsOn(":dist")
     workingDir = rootDir
-    jvmArgs!!.removeIf { it.contains("-Xmx") }
-    maxHeapSize = "3g"
-
     useJUnitPlatform()
 }
 

compiler/fir/analysis-tests/legacy-fir-tests/build.gradle.kts

@@ -47,11 +47,9 @@ if (kotlinBuildProperties.isInJpsBuildIdeaSync) {
     }
 }
 
-projectTest(parallel = true) {
+projectTest(parallel = true, maxHeapSizeMb = 3072) {
     dependsOn(":dist")
     workingDir = rootDir
-    jvmArgs!!.removeIf { it.contains("-Xmx") }
-    maxHeapSize = "3g"
 }
 
 testsJar()

compiler/fir/modularized-tests/build.gradle.kts

@@ -43,12 +43,9 @@ sourceSets {
     "test" { projectDefault() }
 }
 
-projectTest {
+projectTest(minHeapSizeMb = 8192, maxHeapSizeMb = 8192, reservedCodeCacheSizeMb = 512) {
     systemProperties(project.properties.filterKeys { it.startsWith("fir.") })
     workingDir = rootDir
-    jvmArgs!!.removeIf { it.contains("-Xmx") || it.contains("-Xms") || it.contains("ReservedCodeCacheSize") }
-    minHeapSize = "8g"
-    maxHeapSize = "8g"
     dependsOn(":dist")
 
     run {
@@ -58,7 +55,6 @@ projectTest {
             jvmArgs(paramRegex.findAll(argsExt).map { it.groupValues[1] }.toList())
         }
     }
-    jvmArgs("-XX:ReservedCodeCacheSize=512m")
 }
 
 testsJar()

compiler/tests-common-new/build.gradle.kts

@@ -60,9 +60,6 @@ if (kotlinBuildProperties.isInJpsBuildIdeaSync) {
 projectTest(jUnitMode = JUnitMode.JUnit5) {
     dependsOn(":dist")
     workingDir = rootDir
-    jvmArgs!!.removeIf { it.contains("-Xmx") }
-    maxHeapSize = "3g"
-
     useJUnitPlatform()
 }