From 09bca33a8b37820c1eaaa353a52fba779d2c823a Mon Sep 17 00:00:00 2001
From: agibsonccc <adam@konduit.ai>
Date: Fri, 19 Mar 2021 12:27:06 +0900
Subject: [PATCH] Increase ram for builds

---
 .../ArrowWritableRecordTimeSeriesBatchTests.java            | 6 +++---
 .../nd4j-parameter-server-node/pom.xml                      | 4 ++--
 pom.xml                                                     | 1 +
 3 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/datavec/datavec-arrow/src/test/java/org/datavec/arrow/recordreader/ArrowWritableRecordTimeSeriesBatchTests.java b/datavec/datavec-arrow/src/test/java/org/datavec/arrow/recordreader/ArrowWritableRecordTimeSeriesBatchTests.java
index f220288bd..0831073f5 100644
--- a/datavec/datavec-arrow/src/test/java/org/datavec/arrow/recordreader/ArrowWritableRecordTimeSeriesBatchTests.java
+++ b/datavec/datavec-arrow/src/test/java/org/datavec/arrow/recordreader/ArrowWritableRecordTimeSeriesBatchTests.java
@@ -55,9 +55,9 @@ public class ArrowWritableRecordTimeSeriesBatchTests extends BaseND4JTest {
 
 
         List<List<Writable>> timeStep = Arrays.asList(
-                Arrays.<Writable>asList(new IntWritable(0),new IntWritable(1),new IntWritable(2)),
-                Arrays.<Writable>asList(new IntWritable(1),new IntWritable(2),new IntWritable(3)),
-                Arrays.<Writable>asList(new IntWritable(4),new IntWritable(5),new IntWritable(6))
+                Arrays.asList(new IntWritable(0),new IntWritable(1),new IntWritable(2)),
+                Arrays.asList(new IntWritable(1),new IntWritable(2),new IntWritable(3)),
+                Arrays.asList(new IntWritable(4),new IntWritable(5),new IntWritable(6))
         );
 
         int numTimeSteps = 5;
diff --git a/nd4j/nd4j-parameter-server-parent/nd4j-parameter-server-node/pom.xml b/nd4j/nd4j-parameter-server-parent/nd4j-parameter-server-node/pom.xml
index 4eb2a05e2..d2c370aaf 100644
--- a/nd4j/nd4j-parameter-server-parent/nd4j-parameter-server-node/pom.xml
+++ b/nd4j/nd4j-parameter-server-parent/nd4j-parameter-server-node/pom.xml
@@ -111,7 +111,7 @@
                                 <include>*.java</include>
                                 <include>**/*.java</include>
                             </includes>
-                            <argLine> </argLine>
+                            <argLine> -Xmx8g </argLine>
                         </configuration>
                     </plugin>
                 </plugins>
@@ -135,7 +135,7 @@
                         <groupId>org.apache.maven.plugins</groupId>
                         <artifactId>maven-surefire-plugin</artifactId>
                         <configuration>
-                            <argLine> -Djava.library.path="${nd4j.basedir}/nd4j-backends/nd4j-backend-impls/nd4j-cuda/target/classes"</argLine>
+                            <argLine>-Xmx8g</argLine>
                         </configuration>
                     </plugin>
                 </plugins>
diff --git a/pom.xml b/pom.xml
index 5a05d9b97..7690b8b9f 100644
--- a/pom.xml
+++ b/pom.xml
@@ -1166,6 +1166,7 @@
                             <environmentVariables>
                                 <DL4J_INTEGRATION_TESTS>true</DL4J_INTEGRATION_TESTS>
                             </environmentVariables>
+                            <argLine>-Xmx8g</argLine>
                         </configuration>
                         <dependencies>
                             <dependency>