public void syncCublasStream() { if (cublasStream != null) { if (nativeOps.streamSynchronize(cublasStream) == 0) throw new ND4JIllegalStateException("CUDA stream synchronization failed"); } else throw new IllegalStateException("cuBLAS stream isnt set"); }
public void syncSolverStream() { if (solverStream != null) { if (nativeOps.streamSynchronize(solverStream) == 0) throw new ND4JIllegalStateException("CUDA stream synchronization failed"); } else throw new IllegalStateException("cuBLAS stream isnt set"); }
public void syncCublasStream() { if (cublasStream != null) { if (nativeOps.streamSynchronize(cublasStream) == 0) throw new ND4JIllegalStateException("CUDA stream synchronization failed"); } else throw new IllegalStateException("cuBLAS stream isnt set"); }
public void syncSolverStream() { if (solverStream != null) { if (nativeOps.streamSynchronize(solverStream) == 0) throw new ND4JIllegalStateException("CUDA stream synchronization failed"); } else throw new IllegalStateException("cuBLAS stream isnt set"); }
public void syncSpecialStream() { if (nativeOps.streamSynchronize(specialStream) == 0) throw new ND4JIllegalStateException("CUDA special stream synchronization failed"); }
public void syncSpecialStream() { if (nativeOps.streamSynchronize(specialStream) == 0) throw new ND4JIllegalStateException("CUDA special stream synchronization failed"); }
public void syncOldStream(boolean syncCuBlas) { // ContextHolder.getInstance().setContext(); if (nativeOps.streamSynchronize(oldStream) == 0) throw new ND4JIllegalStateException("CUDA stream synchronization failed"); if (syncCuBlas) syncCublasStream(); }
public void syncOldStream(boolean syncCuBlas) { // ContextHolder.getInstance().setContext(); if (nativeOps.streamSynchronize(oldStream) == 0) throw new ND4JIllegalStateException("CUDA stream synchronization failed"); if (syncCuBlas) syncCublasStream(); }
public int synchronize() { NativeOps nativeOps = NativeOpsHolder.getInstance().getDeviceNativeOps(); int res = nativeOps.streamSynchronize(this); if (res == 0) throw new ND4JException("CUDA exception happened. Terminating. Last op: [" + Nd4j.getExecutioner().getLastOp() +"]"); return res; } }
public int synchronize() { NativeOps nativeOps = NativeOpsHolder.getInstance().getDeviceNativeOps(); int res = nativeOps.streamSynchronize(this); if (res == 0) throw new ND4JException("CUDA exception happened. Terminating. Last op: [" + Nd4j.getExecutioner().getLastOp() +"]"); return res; } }