70 lines
1.9 KiB
Julia
70 lines
1.9 KiB
Julia
using ArgParse
|
|
|
|
###############################################################################
|
|
#
|
|
# Parsing command line
|
|
#
|
|
###############################################################################
|
|
|
|
function parse_commandline()
|
|
settings = ArgParseSettings()
|
|
|
|
@add_arg_table settings begin
|
|
"--tol"
|
|
help = "set numerical tolerance for the SDP solver"
|
|
arg_type = Float64
|
|
default = 1e-14
|
|
"--iterations"
|
|
help = "set maximal number of iterations for the SDP solver (default: 20000)"
|
|
arg_type = Int
|
|
default = 60000
|
|
"--upper-bound"
|
|
help = "Set an upper bound for the spectral gap"
|
|
arg_type = Float64
|
|
default = Inf
|
|
"--cpus"
|
|
help = "Set number of cpus used by solver (default: auto)"
|
|
arg_type = Int
|
|
required = false
|
|
"-N"
|
|
help = "Consider automorphisms of free group on N generators"
|
|
arg_type = Int
|
|
default = 2
|
|
"--radius"
|
|
help = "Radius of ball B_r(e,S) to find solution over"
|
|
arg_type = Int
|
|
default = 2
|
|
end
|
|
|
|
return parse_args(settings)
|
|
end
|
|
|
|
parsed_args = parse_commandline()
|
|
|
|
function cpuinfo_physicalcores()
|
|
maxcore = -1
|
|
for line in eachline("/proc/cpuinfo")
|
|
if startswith(line, "core id")
|
|
maxcore = max(maxcore, parse(Int, split(line, ':')[2]))
|
|
end
|
|
end
|
|
maxcore < 0 && error("failure to read core ids from /proc/cpuinfo")
|
|
return maxcore + 1
|
|
end
|
|
|
|
if parsed_args["cpus"] == nothing
|
|
N = cpuinfo_physicalcores()
|
|
parsed_args["cpus"] = N
|
|
info("Setting --cpus to $N")
|
|
elseif parsed_args["cpus"] > cpuinfo_physicalcores()
|
|
warn("Number of specified cores exceeds the physical core count. Performance may suffer.")
|
|
end
|
|
|
|
addprocs(parsed_args["cpus"])
|
|
BLAS.set_num_threads(parsed_args["cpus"])
|
|
|
|
include("SAutFNs.jl")
|
|
|
|
include("Orbit.jl")
|
|
main(SAutFNs, parsed_args)
|