remote/bin/slurm-shell @ 417f91fae3c2

More
author Steve Losh <steve@stevelosh.com>
date Thu, 28 Mar 2024 16:06:55 -0400
parents (none)
children (none)
#!/usr/bin/env bash

set -euo pipefail

read -p "Account? "
acct="$REPLY"

read -p "Hours? "
hours="$REPLY"

read -p "CPUs? "
cpus="$REPLY"

read -p "Memory (gb)? "
mem_gb="$REPLY"
mem_kb=$(( mem_gb * 1024 * 1024 / cpus ))

exec salloc --account="$acct" --nodes=1 --ntasks-per-node=1 --mem-per-cpu="$mem_kb"K --cpus-per-task="$cpus" --time="$hours":00:00