/** * Assert that no workspaces are currently open * * @param msg Message to include in the exception, if required */ public static void assertNoWorkspacesOpen(String msg) throws ND4JWorkspaceException { if (Nd4j.getWorkspaceManager().anyWorkspaceActiveForCurrentThread()) { List<MemoryWorkspace> l = Nd4j.getWorkspaceManager().getAllWorkspacesForCurrentThread(); List<String> workspaces = new ArrayList<>(l.size()); for (MemoryWorkspace ws : l) { if(ws.isScopeActive()) { workspaces.add(ws.getId()); } } throw new ND4JWorkspaceException(msg + " - Open/active workspaces: " + workspaces); } }
if (Nd4j.getWorkspaceManager().anyWorkspaceActiveForCurrentThread()) { val ws = Nd4j.getMemoryManager().getCurrentWorkspace(); nativeOps.memcpyAsync(target.addressPointer(), dstPtr, target.capacity(), CudaConstants.cudaMemcpyHostToHost, stream); if (Nd4j.getWorkspaceManager().anyWorkspaceActiveForCurrentThread()) { if (Nd4j.getWorkspaceManager().anyWorkspaceActiveForCurrentThread()) {