-
Notifications
You must be signed in to change notification settings - Fork 24
Remove aot code #350
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
base: main
Are you sure you want to change the base?
Remove aot code #350
Conversation
Signed-off-by: Ivan Butygin <[email protected]>
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.
Pull Request Overview
This PR removes legacy AOT (Ahead-of-Time) compilation code from the Wave language project. The changes clean up turbine-related code remains after the removal of the main turbine codebase.
- Removes entire dynamo and aot modules containing type conversion, decomposition, and compilation infrastructure
- Eliminates procedural tracing system and IREE emission components
- Adds test requirement for turbine integration
Reviewed Changes
Copilot reviewed 24 out of 24 changed files in this pull request and generated no comments.
Show a summary per file
| File | Description |
|---|---|
| wave_lang/dynamo/type_conversion.py | Removed torch type to native type conversion utilities |
| wave_lang/dynamo/decompositions.py | Removed decomposition management for torch operations |
| wave_lang/dynamo/init.py | Removed empty dynamo module initialization |
| wave_lang/aot/tensor_traits.py | Removed device affinity and external tensor trait classes |
| wave_lang/aot/support/procedural/tracer.py | Removed procedural IR tracing infrastructure |
| wave_lang/aot/support/procedural/primitives.py | Removed IR scalar and tensor primitive types |
| wave_lang/aot/support/procedural/iree_emitter.py | Removed IREE tensor dialect emission API |
| wave_lang/aot/support/procedural/globals.py | Removed global reference management system |
| wave_lang/aot/support/procedural/exported_program.py | Removed exported program import utilities |
| wave_lang/aot/support/procedural/base.py | Removed base procedural tracing classes |
| wave_lang/aot/support/procedural/init.py | Removed procedural package initialization |
| wave_lang/aot/support/ir_utils.py | Removed IR construction and module building utilities |
| wave_lang/aot/passes/functorch.py | Removed torch functionalization passes |
| wave_lang/aot/passes/init.py | Removed passes package initialization |
| wave_lang/aot/params.py | Removed parameter archive management system |
| wave_lang/aot/fx_programs.py | Removed FX program export and serialization |
| wave_lang/aot/exporter.py | Removed main AOT export functionality |
| wave_lang/aot/decompositions.py | Removed AOT decomposition context management |
| wave_lang/aot/compiled_module.py | Removed compiled module class and metadata |
| wave_lang/aot/builtins/jittable.py | Removed JIT compilation decorator |
| wave_lang/aot/builtins/globals.py | Removed global export utilities |
| wave_lang/aot/builtins/init.py | Removed builtins package initialization |
| wave_lang/aot/init.py | Removed aot package initialization |
| lit_tests/kernel/wave/sharktank_integration.py | Added turbine requirement to test |
Tip: Customize your code reviews with copilot-instructions.md. Create the file or learn how to get started.
Signed-off-by: Ivan Butygin <[email protected]>
Signed-off-by: Ivan Butygin <[email protected]>
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.
Pull Request Overview
Copilot reviewed 29 out of 29 changed files in this pull request and generated no new comments.
Tip: Customize your code reviews with copilot-instructions.md. Create the file or learn how to get started.
|
this code may still be used in sharktank repo, check with them before merging |
Remove turbine code remains.