Skip to content
Permalink

Comparing changes

Choose two branches to see what’s changed or to start a new pull request. If you need to, you can also or learn more about diff comparisons.

Open a pull request

Create a new pull request by comparing changes across two branches. If you need to, you can also . Learn more about diff comparisons here.
base repository: johnmarktaylor91/torchlens
Failed to load repositories. Confirm that selected base ref is valid, then try again.
Loading
base: v0.21.0
Choose a base ref
...
head repository: johnmarktaylor91/torchlens
Failed to load repositories. Confirm that selected head ref is valid, then try again.
Loading
compare: v0.21.1
Choose a head ref
  • 5 commits
  • 8 files changed
  • 3 contributors

Commits on Mar 9, 2026

  1. perf(postprocess): optimize pipeline for large models 

    - Per-step verbose timing: unwrap grouped _vtimed blocks into individual
  step timing with graph-stats summary, enabling users to identify which
  specific step is slow (O16)
- Cache module_str by containing_modules tuple to avoid redundant
  string joins in Step 6 (O8)
- Early-continue guards in _undecorate_all_saved_tensors to skip BFS
  on layers with empty captured_args/kwargs (O5)
- Pre-compute buffer_layers_by_module dict in _build_module_logs,
  eliminating O(modules × buffers) scan per module (O6)
- Single-pass arglist rebuild in Step 11 rename, replacing 3-pass
  enumerate + index set + filter pattern (O2)
- Replace OrderedDict with dict in _trim_and_reorder (Python 3.7+
  preserves insertion order) for lower allocation overhead (O4)
- Reverse-index approach in _refine_iso_groups: O(members × neighbors)
  instead of O(members²) all-pairs combinations (O9)
- Pre-compute param types per subgraph as frozenset before pair loop
  in _merge_iso_groups_to_layers (O10)
- Set-based O(n) collision detection replacing O(n²) .count() calls
  in _find_isomorphic_matches (O12)

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
    @johnmarktaylor91 @claude
    johnmarktaylor91 and claude committed Mar 9, 2026
    Configuration menu
    Copy the full SHA
    a211417 View commit details
    Browse the repository at this point in the history

  2. chore: trigger CI 

    Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
    @johnmarktaylor91 @claude
    johnmarktaylor91 and claude committed Mar 9, 2026
    Configuration menu
    Copy the full SHA
    99f4102 View commit details
    Browse the repository at this point in the history

  3. fix(postprocess): fix mypy type errors in _build_module_param_info 

    Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
    @johnmarktaylor91 @claude
    johnmarktaylor91 and claude committed Mar 9, 2026
    Configuration menu
    Copy the full SHA
    11ea006 View commit details
    Browse the repository at this point in the history

  4. Merge pull request #131 from johnmarktaylor91/perf/postprocess-pipeli… 

    …ne-optimizations
    @johnmarktaylor91
    johnmarktaylor91 authored Mar 9, 2026
    Configuration menu
    Copy the full SHA
    69099e8 View commit details
    Browse the repository at this point in the history

  5. chore(release): 0.21.1

    semantic-release committed Mar 9, 2026
    Configuration menu
    Copy the full SHA
    364ea39 View commit details
    Browse the repository at this point in the history
Loading