Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion docs/platforms/aurora_operations.md
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@ cat $PBS_NODEFILE
Open a new terminal on the UAN and run:

```bash
cd /lus/flare/projects/ModCon/ngetty/PRISM
cd /lus/flare/projects/<project>/<user>/PRISM

# Quick test (5 steps, 1B model):
python tools/launch_aurora_web.py \
Expand Down
2 changes: 1 addition & 1 deletion docs/platforms/running_on_polaris.md
Original file line number Diff line number Diff line change
Expand Up @@ -98,7 +98,7 @@ recommended layout:
datasets/<dataset>/shards/*.tar
models/<hf-org--hf-model>/ # HF snapshot dirs
envs/.venv-polaris/ # built by setup_polaris_env.sh
repos/PRISM/ # checkout
repos/PRISM/ # checkout
logs/<DESIGN>/ # PBS stdout/stderr
```

Expand Down
2 changes: 1 addition & 1 deletion tests/test_scits_fix_static.py
Original file line number Diff line number Diff line change
Expand Up @@ -455,7 +455,7 @@ def test_no_user_specific_hub_paths_in_model_configs():

Tokenizer paths under a project dir are still allowed: the interleaved TS
configs legitimately point at custom tokenizers checked in under
/flare/ModCon/<user>/BaseMM_PRISM/tokenizers/.
/flare/<project>/<user>/PRISM/tokenizers/.
"""
offenders = []
for cfg in sorted((REPO_ROOT / "src" / "conf" / "model").glob("*.yaml")):
Expand Down
4 changes: 2 additions & 2 deletions tools/ci/site_path_allowlist.json
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@
"docs/modalities/image_decoder/docci_data.md": 3,
"docs/modalities/timeseries.md": 6,
"docs/models/auroragpt_vlm.md": 1,
"docs/platforms/aurora_operations.md": 4,
"docs/platforms/aurora_operations.md": 3,
"docs/platforms/daos_setup.md": 10,
"docs/platforms/running_on_polaris.md": 8,
"docs/results/patrick_tsqa_ab.md": 3,
Expand Down Expand Up @@ -50,7 +50,7 @@
"tools/isoflop_collect.py": 1,
"tools/isoflop_launch.py": 1,
"tools/isoflop_plan.py": 4,
"tools/launch_aurora.py": 3,
"tools/launch_aurora.py": 2,
"tools/launch_aurora_daos.py": 2,
"tools/launch_aurora_web.py": 1,
"tools/launch_baremetal.py": 2,
Expand Down
5 changes: 4 additions & 1 deletion tools/launch_aurora.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,10 @@ def main():
env_config = load_dotenv()

# Defaults from .env or hardcoded fallbacks
default_prism_dir = env_config.get("PRISM_DIR", "/flare/ModCon/ngetty/BaseMM_PRISM")
# Fall back to the checkout this launcher lives in, not a hardcoded user path.
default_prism_dir = env_config.get(
"PRISM_DIR", os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
)
default_hf_home = env_config.get("HF_HOME", "/flare/ModCon/sandeep")
env_config.get("PROJECT_ALLOCATION", "ModCon")
env_config.get("QUEUE_NAME", "debug")
Expand Down
2 changes: 1 addition & 1 deletion tools/launch_polaris.py
Original file line number Diff line number Diff line change
Expand Up @@ -239,7 +239,7 @@ def main() -> int:
env_config = load_dotenv()

# When --prism-dir isn't given, prefer the launcher's own location over
# cwd. That way `ssh polaris "cd /eagle/.../BaseMM_PRISM && \
# cwd. That way `ssh polaris "cd /eagle/.../PRISM && \
# python tools/launch_polaris.py ..."` Just Works from a different host.
if args.prism_dir:
prism_dir = args.prism_dir
Expand Down
Loading