-
-
Notifications
You must be signed in to change notification settings - Fork 11k
fix test_simple_inductor_graph_partition #26522
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
fix test_simple_inductor_graph_partition #26522
Conversation
Signed-off-by: Boyuan Feng <boyuan@meta.com>
Signed-off-by: Boyuan Feng <boyuan@meta.com>
| compiled_fn = inductor_compiled_graph.current_callable | ||
| file_path = compiled_fn.__code__.co_filename # noqa | ||
| if not file_path.startswith(self.base_cache_dir): | ||
| if ( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
a known cache issue
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
what's the cache issue? do we have an issue on pytorch?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Code Review
This pull request effectively resolves a CI failure in test_simple_inductor_graph_partition by disabling the compilation cache for the test, ensuring each parameterization runs independently. It also proactively adds null checks for compiled_fn.__closure__ in vllm/compilation/compiler_interface.py, enhancing robustness by preventing potential TypeError exceptions. The changes are correct and well-targeted.
Signed-off-by: Boyuan Feng <boyuan@meta.com> Signed-off-by: xuebwang-amd <xuebwang@amd.com>
Signed-off-by: Boyuan Feng <boyuan@meta.com> Signed-off-by: Dhruvil Bhatt <bhattdbh@amazon.com>
Signed-off-by: Boyuan Feng <boyuan@meta.com> Signed-off-by: bbartels <benjamin@bartels.dev>
Signed-off-by: Boyuan Feng <boyuan@meta.com>
Signed-off-by: Boyuan Feng <boyuan@meta.com>
Signed-off-by: Boyuan Feng <boyuan@meta.com> Signed-off-by: xuebwang-amd <xuebwang@amd.com>
Signed-off-by: Boyuan Feng <boyuan@meta.com> Signed-off-by: 0xrushi <6279035+0xrushi@users.noreply.github.com>
Signed-off-by: Boyuan Feng <boyuan@meta.com> Signed-off-by: 0xrushi <6279035+0xrushi@users.noreply.github.com>
Minor fix for a ci failure when enabling inductor graph partition for PyTorch 2.9.
Previously, cache skips re-compilation and cudagraph record, while we assert on the number of cudagraphs recorded.