Brutex
/
cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

disable mmap test for cuda 

Signed-off-by: raver119 <raver119@gmail.com>
master
raver119 2020-04-22 12:57:26 +03:00
parent 1b8077f66a
commit 2c80b18f1d
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/workspace/SpecialWorkspaceTests.java
View File

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