diff --git a/Manifest.toml b/Manifest.toml index 560a983..96e5162 100644 --- a/Manifest.toml +++ b/Manifest.toml @@ -1547,9 +1547,9 @@ version = "0.5.5" [[deps.Oceananigans]] deps = ["Adapt", "Crayons", "CubedSphere", "Dates", "Distances", "DocStringExtensions", "FFTW", "GPUArraysCore", "Glob", "InteractiveUtils", "JLD2", "KernelAbstractions", "Krylov", "LinearAlgebra", "Logging", "MPI", "MuladdMacro", "OffsetArrays", "OrderedCollections", "Pkg", "Printf", "ReactantCore", "Rotations", "SeawaterPolynomials", "SparseArrays", "StaticArrays", "Statistics", "StructArrays"] -git-tree-sha1 = "fdc10bdf6d68bc32bd805ee92fec1dae7f1ff534" +git-tree-sha1 = "2839541ce91b90dd187a9120810b102c34e98a71" uuid = "9e8cae18-63c1-5223-a75c-80ca9d6e9a09" -version = "0.105.0" +version = "0.105.2" [deps.Oceananigans.extensions] OceananigansAMDGPUExt = ["AMDGPU", "AbstractFFTs"] diff --git a/simulations/simulation_runner.py b/simulations/simulation_runner.py index 63f0693..bea147e 100644 --- a/simulations/simulation_runner.py +++ b/simulations/simulation_runner.py @@ -33,7 +33,7 @@ def get_submission_options(scheduler, job_size, gpu_type=None): if scheduler == "pbs": default_gpu_type = "v100" gpu_type_override = gpu_type if gpu_type is not None else default_gpu_type - options = [f"select=1:ncpus=1:ngpus=1:gpu_type={gpu_type_override}:mem=50GB"] + options = [f"select=1:ncpus=8:ngpus=1:gpu_type={gpu_type_override}:mem=50GB"] elif scheduler == "slurm": options = ["--ntasks=1", "--constraint=gpu", @@ -45,7 +45,7 @@ def get_submission_options(scheduler, job_size, gpu_type=None): if scheduler == "pbs": default_gpu_type = "cc80" # Compute capability 8.0 gpu_type_override = gpu_type if gpu_type is not None else default_gpu_type - options = [f"select=1:ncpus=1:ngpus=1:gpu_type={gpu_type_override}:mem=200GB"] + options = [f"select=1:ncpus=8:ngpus=1:gpu_type={gpu_type_override}:mem=200GB"] elif scheduler == "slurm": options = ["--ntasks=1", "--constraint=gpu", @@ -57,7 +57,7 @@ def get_submission_options(scheduler, job_size, gpu_type=None): if scheduler == "pbs": default_gpu_type = "h100" # H100 GPUs specifically gpu_type_override = gpu_type if gpu_type is not None else default_gpu_type - options = [f"select=1:ncpus=1:ngpus=1:gpu_type={gpu_type_override}:mem=200GB", + options = [f"select=1:ncpus=8:ngpus=1:gpu_type={gpu_type_override}:mem=200GB", "job_priority=regular"] elif scheduler == "slurm": options = ["--ntasks=1",