Increase graph ylim if needed

Author: christiangnrd

dev/flopscomp.jl

@@ -13,7 +13,7 @@ using Plots
 using Plots.Measures
 
 n_gpu_cores = "??"
-# Comment this out if scary. Please mention it in your comment when uploading the figure
+# Comment this out if scary. Please mention number of cores in your comment when uploading the figure
 system_prof = read(`system_profiler SPDisplaysDataType`, String)
 n_gpu_cores = only(match(r"Total Number of Cores:\s*(\d+)", system_prof).captures)
 
@@ -146,13 +146,19 @@ function main(; Ns=[50, 64, 100, 128, 250, 256, 500, 512, 1000, 1024, 2000, 2048
         ]
 
     res = Dict()
+
+    ylim_upper = 9e12
+
     for (inT, outT) in Ts
         tmpres = compare(Ns, Fs, inT, outT; n_batch, ntrials)
 
         plt = plot(xlabel="N, n_batch=$(n_batch)", legendtitle="($inT, $outT)")
         for (res, (_, info_str)) in zip(tmpres,Fs)
             flops = tmpres[info_str]
             peakf = @sprintf("%.3e", maximum(flops))
+            if peakf > ylim_upper
+                ylim_upper = peakf * 1.02
+            end
             plot!(plt, Ns, tmpres[info_str]; linewidth=1.5, label="$(peakf) peak: $info_str")
         end
         res[(inT,outT)] = (plt=plt, results=tmpres)