#!/bin/bash
# Produce the vonkarman-modal chain for cymbal-flat-disk-v1. Run locally, never on the vault runner.
# The 8 mm members run on a CPU (minutes); the 4 mm members need a GPU (--device cuda, PyTorch) to be
# practical, since the per-step cost is the memory bandwidth of the coupling tensor (1.7 GB at 600 modes).
# Usage: bash run_chain.sh <cymbal-repo>/challenges/cymbal-flat-disk-v1/challenge.json [gpu|cpu]   (cpu: hours for the 4 mm members)
set -e
here="$(cd "$(dirname "$0")" && pwd)"
ch="$1"
what="${2:-all}"
out="$here/../../solutions/cymbal-flat-disk-v1/vonkarman-modal"
mkdir -p "$out"
cd "$here"
dev=""
[ "$what" = cpu ] || dev="--device cuda"
python3 vk_modal_disk.py "$ch" "$out" $dev --fields g4-w300-a300-m2 g8-w100-a100-m2 g8-w200-a200-m2 g4-w200-a200-m2 g4-w300-a300-m2
python3 make_submission.py "$out" g8-w100-a100-m2 g8-w200-a200-m2 g4-w200-a200-m2 g4-w300-a300-m2