MPS CI Support#1278

Open

huseyincavusbi wants to merge 16 commits intoTransformerLensOrg:devfrom

huseyincavusbi:feat/mps-ci-support

Commits on Apr 20, 2026

Fix type of HookedTransformerConfig.device (#1230)

brendanlong
and
jlarson4
authored

Commits on Apr 30, 2026

Merge pull request #1277 from TransformerLensOrg/dev
jlarson4
authored

Commits on May 2, 2026

feat: Add MPS CI runner support (#1264)
huseyincavusbi
committed
ci: Enable runs on feature branch
huseyincavusbi
committed
fix: Skip heavy model_bridge unit tests on MPS runner due to memory limits
huseyincavusbi
committed
fix: Ignore flaky grouped query attention tests on Mac runner
huseyincavusbi
committed
style: Run make format on test_mps_basic.py
huseyincavusbi
committed
style: Standardize MPS step naming convention in CI
huseyincavusbi
committed
ci: Revert MPS trigger to run only on main PRs and pushes
huseyincavusbi
committed
ci: Remove feature branch from global workflow triggers
huseyincavusbi
committed
fix: Restore torch.device return type in get_device for API stability
huseyincavusbi
committed
docs: Align train and device config docstrings with implementation
huseyincavusbi
committed
fix: Update device type hints to Union[str, torch.device] for consistency
huseyincavusbi
committed
ci: Update mps-checks trigger to include dev branch
huseyincavusbi
committed
test: Update device tests for torch.device compatibility and robustness
huseyincavusbi
committed
ci: Restrict MPS trigger to main branch only
huseyincavusbi
committed