/ challenges-old / cymbal-solutions
Sign in
challenges-old / cymbal-solutions
cymbal-solutions / solvers / vonkarman-modal-disk / run_chain.sh
19 lines · 1.0 KBCodeBlameHistory
f124192Chain procedure: 8 mm on CPU, 4 mm on GPU up to 600 modesJeremy Magland 2# Produce the vonkarman-modal chain for cymbal-flat-disk-v1. Run locally, never on the vault runner.
3# The 8 mm members run on a CPU (minutes); the 4 mm members need a GPU (--device cuda, PyTorch) to be
4# practical, since the per-step cost is the memory bandwidth of the coupling tensor (1.7 GB at 600 modes).
5# Usage: bash run_chain.sh <cymbal-repo>/challenges/cymbal-flat-disk-v1/challenge.json [cpu|gpu|all]
7here="$(cd "$(dirname "$0")" && pwd)"
8ch="$1"
af2a046Von Karman modal solver on a Cartesian grid masked to the disk, site, and workflowJeremy Magland 10out="$here/../../solutions/cymbal-flat-disk-v1/vonkarman-modal"
11mkdir -p "$out"
12cd "$here"
f124192Chain procedure: 8 mm on CPU, 4 mm on GPU up to 600 modesJeremy Magland 13if [ "$what" = cpu ] || [ "$what" = all ]; then
14 python3 vk_modal_disk.py "$ch" "$out" g8-w100-a100-m1 g8-w200-a200-m1
15fi
16if [ "$what" = gpu ] || [ "$what" = all ]; then
17 python3 vk_modal_disk.py "$ch" "$out" --device cuda --fields g4-w600-a600-m4 g4-w200-a200-m2 g4-w300-a300-m2 g4-w450-a450-m2 g4-w600-a600-m4
18fi
19python3 make_submission.py "$out" g8-w100-a100-m1 g8-w200-a200-m1 g4-w200-a200-m2 g4-w300-a300-m2 g4-w450-a450-m2 g4-w600-a600-m4
↑↓ move↵openescclose