Skip to content

fix alg_ext torch compile issue#1540

Merged
wenhuach21 merged 12 commits intomainfrom
fix_alg_ext_compile
Mar 13, 2026
Merged

fix alg_ext torch compile issue#1540
wenhuach21 merged 12 commits intomainfrom
fix_alg_ext_compile

Conversation

@wenhuach21
Copy link
Contributor

@wenhuach21 wenhuach21 commented Mar 12, 2026

Description

Please briefly describe your main changes, the motivation.

Type of Change

  • Bug fix
  • New feature
  • Documentation update
  • Performance improvement
  • Code refactoring
  • Other (please specify):

Related Issues

Fixes or relates to #

Checklist Before Submitting

  • My code has been tested locally.
  • Documentation has been updated as needed.
  • New or updated tests are included where applicable.

Copilot AI review requested due to automatic review settings March 12, 2026 11:32
Copy link
Contributor

Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

This PR aims to resolve a torch.compile-related failure when using alg_ext, by changing how the compressor wires up its block forward function during initialization.

Changes:

  • Remove unused imports (re, safe_device_move_with_meta_handling) from BaseCompressor.
  • Stop compiling block_forward and always use the uncompiled block_forward implementation (previous compile path is commented out).

You can also share your feedback on Copilot code review. Take the survey.

@wenhuach21 wenhuach21 requested review from n1ck-guo and xin3he March 12, 2026 11:39
@wenhuach21 wenhuach21 merged commit 9d54ddc into main Mar 13, 2026
29 checks passed
@wenhuach21 wenhuach21 deleted the fix_alg_ext_compile branch March 13, 2026 10:32
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants