disable mmap test for cuda

Signed-off-by: raver119 <raver119@gmail.com>
2020-04-22 12:57:26 +03:00 · 2020-04-22 12:57:26 +03:00 · 2c80b18f1d
commit 2c80b18f1d
parent 1b8077f66a
1 changed files with 3 additions and 0 deletions
--- a/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/workspace/SpecialWorkspaceTests.java
+++ b/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/workspace/SpecialWorkspaceTests.java
@ -336,6 +336,9 @@ public class SpecialWorkspaceTests extends BaseNd4jTest {

    @Test
    public void testMmapedWorkspaceLimits_1() throws Exception {
+        if (!Nd4j.getEnvironment().isCPU())
+            return;
+
        val tmpFile = Files.createTempFile("some", "file");
        val mmap = WorkspaceConfiguration.builder()
                .initialSize(200 * 1024L * 1024L) // 200mbs