Skip to content

Fix TorchDynamo Cache Issue#1501

Merged
toyxu merged 5 commits intomainfrom
meng_fix_ut
Apr 17, 2025
Merged

Fix TorchDynamo Cache Issue#1501
toyxu merged 5 commits intomainfrom
meng_fix_ut

Conversation

@chunhuanMeng
Copy link
Copy Markdown
Contributor

@chunhuanMeng chunhuanMeng commented Mar 24, 2025

Resolved failures in test_autocast cases caused by TorchDynamo's compilation cache limit. When deterministic mode is enabled, Python-implemented ops are compiled using TorchDynamo instead of their C++ counterparts. Added torch._dynamo.reset() to clear the cache and allow recompilation.

Reference: torchvision/ops/roi_align.py#L256

Copy link
Copy Markdown
Contributor

@frost-intel frost-intel left a comment

Choose a reason for hiding this comment

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

LGTM.

@toyxu toyxu added this pull request to the merge queue Apr 17, 2025
Merged via the queue into main with commit ac60f9b Apr 17, 2025
5 of 7 checks passed
@toyxu toyxu deleted the meng_fix_ut branch April 17, 2025 05:39
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