Skip to content

Add ONNX-MLIR dialect support - #40

Open
kartikeyporwal wants to merge 18 commits into
nebuly-ai:mainfrom
kartikeyporwal:feat/add_onnxmlir
Open

Add ONNX-MLIR dialect support#40
kartikeyporwal wants to merge 18 commits into
nebuly-ai:mainfrom
kartikeyporwal:feat/add_onnxmlir

Conversation

@kartikeyporwal

@kartikeyporwal kartikeyporwal commented Apr 27, 2022

Copy link
Copy Markdown
Contributor

This PR adds the ONNX-MLIR support in the nebullvm based on #31.

Few notable points:

  • The project ONNX-MLIR needs to be built from source
  • It depends on the LLVM-PROJECT that needs to be built from source
  • The python wrapper named PyRuntime is built using pybind11 needs to be in the path

@kartikeyporwal kartikeyporwal changed the title - Add ONNX-MLIR dialect support Add ONNX-MLIR dialect support Apr 27, 2022
@kartikeyporwal
kartikeyporwal marked this pull request as ready for review April 27, 2022 00:32
@diegofiori
diegofiori self-requested a review April 27, 2022 18:18
Comment thread nebullvm/base.py Outdated
Comment thread nebullvm/inference_learners/onnx_mlir.py Outdated
Comment thread nebullvm/inference_learners/onnx_mlir.py Outdated
Comment thread nebullvm/inference_learners/onnx_mlir.py Outdated
Comment thread nebullvm/optimizers/multi_compiler.py Outdated
Comment thread nebullvm/inference_learners/onnx_mlir.py Outdated
@diegofiori

Copy link
Copy Markdown
Collaborator

@kartikeyporwal 🥇🚀 thanks for the amazing contribution ⭐! However, there are a few things to change before merging the code to master (See specific comments in the code section). BTW, we're working on a huge release for next week, so I won't merge this PR before it.

@kartikeyporwal
kartikeyporwal requested a review from diegofiori May 8, 2022 10:44
@kartikeyporwal

Copy link
Copy Markdown
Contributor Author

@Morgoth95 Thanks for suggesting the changes. I've made the necessary changes. Please feel free to review the changes again. Thanks

kartikeyporwal and others added 9 commits May 8, 2022 16:28
* add half precision and transformations logic

* fix bugs

* add support to gpu

* add support to gpu

* fix minor bug in gpu code

* fix minor bug in gpu code

* refactor quantization

* add dataset interface

* fix bug

* fix bugs with dataset api

* fix bug

* update test

* solve minor issues

* fix error in cuda

* fix error with tensorRT

* fix bug in tvm and change name of quantization_ths

* add resources

* Modify readme (nebuly-ai#48)

* Create section - integration with other libraries

* update readme (work-in-progress)

* Minor Readme update

* rename notebook

* update notebook

* rename notebook

* Rename notebook

* Rename notebook

* Rename notebook

* Rename notebook

* Rename notebook

* update version to 0.3.0

* Update readme with latest release information

* Benchmarks

* Update readme

* Update readme with benchmarks

* Update readme, minor changes

* solve api issue with tf

* fix typos in benchmarks

Co-authored-by: morgoth95 <diego.fiori@epfl.ch>
Co-authored-by: Emile Courthoud <59367323+emilecourthoud@users.noreply.github.com>
Co-authored-by: Nebuly <83510798+nebuly-ai@users.noreply.github.com>
@kartikeyporwal

Copy link
Copy Markdown
Contributor Author

Hi team,

Could please help me knowing where does we stand in merging this PR in nebullvm?

Thanks

CC: @Morgoth95

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.

2 participants