Flush stdout so outputs are printed to file more often (#54)

Author: mtfishman

.github/workflows/tests.yml

@@ -12,7 +12,7 @@ jobs:
     strategy:
       matrix:
         version:
-          - '1.3'
+          - '1.6'
           - '1'
         os:
           - ubuntu-latest
@@ -24,8 +24,8 @@ jobs:
           # MacOS not available on x86
           - {os: 'macOS-latest', arch: 'x86'}
           # Only test all os on the latest release
-          - {version: '1.3', os: 'windows-latest'}
-          - {version: '1.3', os: 'macOS-latest'}
+          - {version: '1.6', os: 'windows-latest'}
+          - {version: '1.6', os: 'macOS-latest'}
     steps:
       - uses: actions/checkout@v2
       - uses: julia-actions/setup-julia@latest

Project.toml

@@ -18,14 +18,14 @@ Requires = "ae029012-a4dd-5104-9daa-d747884805df"
 [compat]
 Compat = "3"
 HDF5 = "0.15, 0.16"
-ITensors = "0.2, 0.3"
+ITensors = "0.3"
 Infinities = "0.1"
 IterTools = "1"
 KrylovKit = "0.5"
 OffsetArrays = "1"
 QuadGK = "2"
 Requires = "1"
-julia = "1"
+julia = "1.6"
 
 [extras]
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"

src/subspace_expansion.jl

@@ -158,6 +158,8 @@ function subspace_expansion(
     println(
       "Current bond dimension at bond $b is $dˡ while desired maximum dimension is $maxdim, skipping bond dimension increase",
     )
+    flush(stdout)
+    flush(stderr)
     return (ψ.AL[n1], ψ.AL[n2]), ψ.C[n1], (ψ.AR[n1], ψ.AR[n2])
   end
   maxdim -= dˡ
@@ -174,6 +176,8 @@ function subspace_expansion(
     println(
       "Impossible to do a subspace expansion, probably due to conservation constraints"
     )
+    flush(stdout)
+    flush(stderr)
     return (ψ.AL[n1], ψ.AL[n2]), ψ.C[n1], (ψ.AR[n1], ψ.AR[n2])
   end
 

src/vumps_localham.jl

@@ -555,8 +555,11 @@ function tdvp(
   N = nsites(ψ)
   (ϵᴸ!) = fill(tol, nsites(ψ))
   (ϵᴿ!) = fill(tol, nsites(ψ))
-  outputlevel > 0 &&
+  if outputlevel > 0
     println("Running VUMPS with multisite_update_alg = $multisite_update_alg")
+    flush(stdout)
+    flush(stderr)
+  end
   for iter in 1:maxiter
     ψ, (eᴸ, eᴿ) = tdvp_iteration(
       solver,
@@ -570,13 +573,19 @@ function tdvp(
     )
     ϵᵖʳᵉˢ = max(maximum(ϵᴸ!), maximum(ϵᴿ!))
     maxdimψ = maxlinkdim(ψ[0:(N + 1)])
-    outputlevel > 0 && println(
-      "VUMPS iteration $iter (out of maximum $maxiter). Bond dimension = $maxdimψ, energy = $((eᴸ, eᴿ)), ϵᵖʳᵉˢ = $ϵᵖʳᵉˢ, tol = $tol",
-    )
+    if outputlevel > 0
+      println(
+        "VUMPS iteration $iter (out of maximum $maxiter). Bond dimension = $maxdimψ, energy = $((eᴸ, eᴿ)), ϵᵖʳᵉˢ = $ϵᵖʳᵉˢ, tol = $tol",
+      )
+      flush(stdout)
+      flush(stderr)
+    end
     if ϵᵖʳᵉˢ < tol
       println(
         "Precision error $ϵᵖʳᵉˢ reached tolerance $tol, stopping VUMPS after $iter iterations (of a maximum $maxiter).",
       )
+      flush(stdout)
+      flush(stderr)
       break
     end
   end
@@ -599,16 +608,22 @@ function vumps(
 )
   @assert isinf(time_step) && time_step < 0
   println("Using VUMPS solver with time step $time_step")
+  flush(stdout)
+  flush(stderr)
   return tdvp(vumps_solver, args...; time_step=time_step, solver_tol=solver_tol, kwargs...)
 end
 
 function tdvp(args...; time_step, solver_tol=(x -> x / 100), kwargs...)
   solver = if !isinf(time_step)
     println("Using TDVP solver with time step $time_step")
+    flush(stdout)
+    flush(stderr)
     tdvp_solver
   elseif time_step < 0
     # Call VUMPS instead
     println("Using VUMPS solver with time step $time_step")
+    flush(stdout)
+    flush(stderr)
     vumps_solver
   else
     error("Time step $time_step not supported.")

src/vumps_mpo.jl

@@ -146,6 +146,8 @@ function left_environment(H::InfiniteMPOMatrix, ψ::InfiniteCanonicalMPS; tol=1e
         Ls[1][n], info = linsolve(A, Ls[1][n], 1, -1; tol=tol)
       else
         println("Not implemented")
+        flush(stdout)
+        flush(stderr)
       end
     end
   end
@@ -300,6 +302,8 @@ function right_environment(H::InfiniteMPOMatrix, ψ::InfiniteCanonicalMPS; tol=1
         Rs[1][n], info = linsolve(A, Rs[1][n], 1, -1; tol=tol)
       else
         println("Not yet implemented")
+        flush(stdout)
+        flush(stderr)
       end
     end
   end