Skip to content

[frontend] Add MLIR result checking and JIT encapsulation#686

Closed
sdh1014 wants to merge 6 commits intobuddy-compiler:mainfrom
sdh1014:feat/add-mlir-res-check
Closed

[frontend] Add MLIR result checking and JIT encapsulation#686
sdh1014 wants to merge 6 commits intobuddy-compiler:mainfrom
sdh1014:feat/add-mlir-res-check

Conversation

@sdh1014
Copy link
Contributor

@sdh1014 sdh1014 commented Feb 2, 2026

Summary

  • Add a unified JIT wrapper to drive frontend import, compilation, and execution consistently.
  • Introduce numeric correctness validation for ATen lowering and fix multiple graph-level semantic issues.
  • Validate core ATen IR numerics to ensure semantic correctness of the core pipeline.
  • Add a runtime/ component that provides RNG support via C++ external calls for random-related ops.
  • Enable complex64 support to cover selected complex FFT-related operators.
    You can verify this change by running the following commands.
cd build
export BUDDY_MLIR_BUILD_DIR=$PWD
export LLVM_MLIR_BUILD_DIR=$PWD/../llvm/build
export PYTHONPATH=${LLVM_MLIR_BUILD_DIR}/tools/mlir/python_packages/mlir_core:${BUDDY_MLIR_BUILD_DIR}/python_packages:${PYTHONPATH}
cd ..
# Core Aten Ops
python3 tests/Python/AtenOps/run_numeric.py

TODO:The numerical verification for existing Aten Ops from torch.ops.aten has not yet been fully completed.

Checklist

  • The code builds successfully
  • Existing tests pass
  • New tests are added where appropriate
  • Code follows the project coding style
  • Documentation is updated if needed

@sdh1014 sdh1014 requested a review from zhanghb97 as a code owner February 2, 2026 01:09
@sdh1014 sdh1014 closed this Feb 11, 2026
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.

1 participant