@Override public MemoryWorkspace getWorkspaceForCurrentThread(@NonNull WorkspaceConfiguration configuration, @NonNull String id) { ensureThreadExistense(); MemoryWorkspace workspace = backingMap.get().get(id); if (workspace == null) { workspace = newWorkspace(configuration, id); backingMap.get().put(id, workspace); if (Nd4j.getWorkspaceManager().getDebugMode() != DebugMode.BYPASS_EVERYTHING) pickReference(workspace); } return workspace; }
@Override public MemoryWorkspace createNewWorkspace(WorkspaceConfiguration configuration, String id) { ensureThreadExistense(); MemoryWorkspace workspace = newWorkspace(configuration, id); backingMap.get().put(id, workspace); if (Nd4j.getWorkspaceManager().getDebugMode() != DebugMode.BYPASS_EVERYTHING) pickReference(workspace); return workspace; }
@Override public MemoryWorkspace createNewWorkspace(@NonNull WorkspaceConfiguration configuration) { ensureThreadExistense(); MemoryWorkspace workspace = newWorkspace(configuration); backingMap.get().put(workspace.getId(), workspace); if (Nd4j.getWorkspaceManager().getDebugMode() != DebugMode.BYPASS_EVERYTHING) pickReference(workspace); return workspace; }
@Override public MemoryWorkspace createNewWorkspace(WorkspaceConfiguration configuration, String id, Integer deviceId) { ensureThreadExistense(); MemoryWorkspace workspace = newWorkspace(configuration, id, deviceId); backingMap.get().put(id, workspace); if (Nd4j.getWorkspaceManager().getDebugMode() != DebugMode.BYPASS_EVERYTHING) pickReference(workspace); return workspace; }
@Override public MemoryWorkspace createNewWorkspace() { ensureThreadExistense(); MemoryWorkspace workspace = newWorkspace(defaultConfiguration); backingMap.get().put(workspace.getId(), workspace); if (Nd4j.getWorkspaceManager().getDebugMode() != DebugMode.BYPASS_EVERYTHING) pickReference(workspace); return workspace; }
protected MemoryWorkspace newWorkspace(WorkspaceConfiguration configuration, String id) { return Nd4j.getWorkspaceManager().getDebugMode() == DebugMode.BYPASS_EVERYTHING ? new DummyWorkspace() : new CudaWorkspace(configuration, id); }
protected MemoryWorkspace newWorkspace(WorkspaceConfiguration configuration) { return Nd4j.getWorkspaceManager().getDebugMode() == DebugMode.BYPASS_EVERYTHING ? new DummyWorkspace() : new CudaWorkspace(configuration); }
protected MemoryWorkspace newWorkspace(WorkspaceConfiguration configuration, String id, int deviceId) { return Nd4j.getWorkspaceManager().getDebugMode() == DebugMode.BYPASS_EVERYTHING ? new DummyWorkspace() : new CudaWorkspace(configuration, id, deviceId); }
if (deviceOffset.get() + requiredMemory <= currentSize.get() && !trimmer && Nd4j.getWorkspaceManager().getDebugMode() != DebugMode.SPILL_EVERYTHING) { cycleAllocations.addAndGet(requiredMemory); long prevOffset = deviceOffset.getAndAdd(requiredMemory); if (workspaceConfiguration.getPolicyReset() == ResetPolicy.ENDOFBUFFER_REACHED && currentSize.get() > 0 && !trimmer && Nd4j.getWorkspaceManager().getDebugMode() != DebugMode.SPILL_EVERYTHING) { if (hostOffset.get() + requiredMemory <= currentSize.get() && !trimmer && Nd4j.getWorkspaceManager().getDebugMode() != DebugMode.SPILL_EVERYTHING) {