Repository navigation

feat(chs): port pinned memory host-device transfer benchmark to chs - #143

Merged

linamy85 merged 7 commits into

AI-Hypercomputer:chsfrom

simrankaurb:feat/chs-pinned-memory-host-device

Oct 23, 2026

Merged

feat(chs): port pinned memory host-device transfer benchmark to chs#143
linamy85 merged 7 commits into
AI-Hypercomputer:chsfrom
simrankaurb:feat/chs-pinned-memory-host-device

Commits on Oct 21, 2026
Commits on Oct 22, 2026
- fix(chs): pin jax and jaxlib to 0.11.1 with libtpu 0.0.48 to avoid D2H/H2D regression
  simrankaurb
  committed
  8c2fad7
Commits on Oct 23, 2026
- fix(chs): assert device count and pin jax 0.12.0 / libtpu 0.0.50
  simrankaurb
  committed
  695dc7f
- fix(chs): use bare jax==0.11.1 to avoid libtpu dependency conflict
  simrankaurb
  committed
  d7867ff