Jackoatmon commited on
Commit
3af16fd
·
verified ·
1 Parent(s): ddf61f4

Update Feather h200 training runtime image

Browse files
BUILD_STAMP.txt CHANGED
@@ -1 +1 @@
1
- git=c51880dbba451428c88c3244a0bf311995efbfbc time=2026-05-08T14:55:54Z purpose=h200-htm-micro-canary pathfix=c51880db
 
1
+ git=cab0c6eeee8b3d69f3627bfbf80efee5bf9e8eba time=2026-05-08T15:05:02Z purpose=h200-htm-micro-canary importfix=cab0c6ee
overlay/scripts/htm_gpu_micro_canary.py CHANGED
@@ -11,10 +11,27 @@ from __future__ import annotations
11
  import argparse
12
  import json
13
  import os
14
- import random
15
  import time
 
 
16
  from typing import Any
17
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
18
 
19
  def build_htm_env(mode: str) -> dict[str, str]:
20
  """Return env overrides for the requested HTM diagnostic mode."""
@@ -86,6 +103,7 @@ def main(argv: list[str] | None = None) -> int:
86
  return 0
87
 
88
  import torch
 
89
  from subsystems.htm import HTMLayer
90
 
91
  emit(
 
11
  import argparse
12
  import json
13
  import os
14
+ import sys
15
  import time
16
+ import traceback
17
+ from pathlib import Path
18
  from typing import Any
19
 
20
+ import torch
21
+
22
+
23
+ def ensure_repo_on_path() -> None:
24
+ """Make overlay package imports work from both /app/scripts and repo-root runs."""
25
+ candidates = [
26
+ Path('/workspace/feather'),
27
+ Path(__file__).resolve().parents[1] if len(Path(__file__).resolve().parents) > 1 else None,
28
+ ]
29
+ for candidate in candidates:
30
+ if candidate and (candidate / 'subsystems' / 'htm.py').exists():
31
+ candidate_s = str(candidate)
32
+ if candidate_s not in sys.path:
33
+ sys.path.insert(0, candidate_s)
34
+ return
35
 
36
  def build_htm_env(mode: str) -> dict[str, str]:
37
  """Return env overrides for the requested HTM diagnostic mode."""
 
103
  return 0
104
 
105
  import torch
106
+ ensure_repo_on_path()
107
  from subsystems.htm import HTMLayer
108
 
109
  emit(