Skip to content
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.

Sign up for GitHub

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

Jump to bottom

Demonstrate how to add JIT using MLIR to micrograd #62

Open
wants to merge 25 commits into
base: master
Choose a base branch
from
Open

Demonstrate how to add JIT using MLIR to micrograd #62

wants to merge 25 commits into from

Commits on Jan 12, 2024

  1. ignore venv

    @fzakaria
    fzakaria committed Jan 12, 2024
    Configuration menu
    Copy the full SHA
    078afbc View commit details
    Browse the repository at this point in the history

Commits on Jan 20, 2024

  1. Added more to gitignore

    @fzakaria
    fzakaria committed Jan 20, 2024
    Configuration menu
    Copy the full SHA
    23c7fac View commit details
    Browse the repository at this point in the history

  2. Minor fixes to engine.py for passing parent information for __pow__

    @fzakaria
    fzakaria committed Jan 20, 2024
    Configuration menu
    Copy the full SHA
    edcde3d View commit details
    Browse the repository at this point in the history

  3. Add a visitor

    @fzakaria
    fzakaria committed Jan 20, 2024
    Configuration menu
    Copy the full SHA
    f1cab09 View commit details
    Browse the repository at this point in the history

  4. Add a test visitor

    @fzakaria
    fzakaria committed Jan 20, 2024
    Configuration menu
    Copy the full SHA
    b4cbbcd View commit details
    Browse the repository at this point in the history

Commits on Feb 28, 2024

  1. Add test_mlir_execution.py

    @alexander-shaposhnikov
    alexander-shaposhnikov committed Feb 28, 2024
    Configuration menu
    Copy the full SHA
    ff2593b View commit details
    Browse the repository at this point in the history

  2. Simplify the pipeline

    @alexander-shaposhnikov
    alexander-shaposhnikov committed Feb 28, 2024
    Configuration menu
    Copy the full SHA
    621328a View commit details
    Browse the repository at this point in the history

Commits on Feb 29, 2024

  1. Add jit.py, test_jit.py

    @alexander-shaposhnikov
    alexander-shaposhnikov committed Feb 29, 2024
    Configuration menu
    Copy the full SHA
    25138a0 View commit details
    Browse the repository at this point in the history

  2. Merge pull request #1 from alexander-shaposhnikov/add_mlir_exec 

    Add test_mlir_execution.py
    @fzakaria
    fzakaria authored Feb 29, 2024
    Configuration menu
    Copy the full SHA
    67516d4 View commit details
    Browse the repository at this point in the history

  3. Merge pull request #2 from alexander-shaposhnikov/add_compiler 

    Add jit.py, test_jit.py
    @fzakaria
    fzakaria authored Feb 29, 2024
    Configuration menu
    Copy the full SHA
    574f731 View commit details
    Browse the repository at this point in the history

  4. Add direnv integration

    @fzakaria
    fzakaria committed Feb 29, 2024
    Configuration menu
    Copy the full SHA
    fc6c096 View commit details
    Browse the repository at this point in the history

  5. Add requirements.txt file

    @fzakaria
    fzakaria committed Feb 29, 2024
    Configuration menu
    Copy the full SHA
    b27f24d View commit details
    Browse the repository at this point in the history

  6. Added numpy to requirements

    @fzakaria
    fzakaria committed Feb 29, 2024
    Configuration menu
    Copy the full SHA
    518e151 View commit details
    Browse the repository at this point in the history

  7. Add __init__.py to test directory

    @fzakaria
    fzakaria committed Feb 29, 2024
    Configuration menu
    Copy the full SHA
    440f14a View commit details
    Browse the repository at this point in the history

  8. Fixup some tests to test against non JIT

    @fzakaria
    fzakaria committed Feb 29, 2024
    Configuration menu
    Copy the full SHA
    0bfcaf3 View commit details
    Browse the repository at this point in the history

  9. Changed README for how to invoke pytest

    @fzakaria
    fzakaria committed Feb 29, 2024
    Configuration menu
    Copy the full SHA
    2213aea View commit details
    Browse the repository at this point in the history

  10. Added a JIT callable to print mlir

    @fzakaria
    fzakaria committed Feb 29, 2024
    Configuration menu
    Copy the full SHA
    84cb555 View commit details
    Browse the repository at this point in the history

  11. Refinements 

    * Added JIT doc to README
* Added more comments
* Cleaned up ctypes code
    @fzakaria
    fzakaria committed Feb 29, 2024
    Configuration menu
    Copy the full SHA
    87489d7 View commit details
    Browse the repository at this point in the history

Commits on Mar 1, 2024

  1. Fix it so that JIT works for multiple out NN 

    * Crazy amount of debugging to fix this. Alexander went crazy deep to
  look at the instructions in lldb to see that the MLIR CPU runner was
  doing a double dereference only when the return is a list.

  The argument is also first in the method list ... why!?

Co-authored-by: Alexander Shaposhnikov <ashaposhnikov@google.com>
    @fzakaria @alexander-shaposhnikov
    fzakaria and alexander-shaposhnikov committed Mar 1, 2024
    Configuration menu
    Copy the full SHA
    0082cc0 View commit details
    Browse the repository at this point in the history

  2. Remove extra argument to execution_engine

    @fzakaria
    fzakaria committed Mar 1, 2024
    Configuration menu
    Copy the full SHA
    c64757b View commit details
    Browse the repository at this point in the history

  3. Add documentation

    @fzakaria
    fzakaria committed Mar 1, 2024
    Configuration menu
    Copy the full SHA
    284c2f1 View commit details
    Browse the repository at this point in the history

  4. Added a new cell to the demo

    @fzakaria
    fzakaria committed Mar 1, 2024
    Configuration menu
    Copy the full SHA
    e79081b View commit details
    Browse the repository at this point in the history

Commits on Mar 2, 2024

  1. Add a toy benchmark

    @alexander-shaposhnikov
    alexander-shaposhnikov committed Mar 2, 2024
    Configuration menu
    Copy the full SHA
    0b5ac71 View commit details
    Browse the repository at this point in the history

  2. Merge pull request #4 from alexander-shaposhnikov/add_toy_benchmark 

    Add a toy benchmark
    @fzakaria
    fzakaria authored Mar 2, 2024
    Configuration menu
    Copy the full SHA
    e8b5d5a View commit details
    Browse the repository at this point in the history

  3. Format test_jit.py

    @fzakaria
    fzakaria committed Mar 2, 2024
    Configuration menu
    Copy the full SHA
    d3e5e4d View commit details
    Browse the repository at this point in the history