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

Add support for calling Triton code from NineToothed and 0-dimensional tensors #9

Merged
merged 7 commits into from
Oct 22, 2024
Merged

Add support for calling Triton code from NineToothed and 0-dimensional tensors #9

merged 7 commits into from
Oct 22, 2024

Commits on Oct 21, 2024

  1. Store the generated source code in jit._Handle for debugging purposes

    @voltjia
    voltjia committed Oct 21, 2024
    Configuration menu
    Copy the full SHA
    b3edc8d View commit details
    Browse the repository at this point in the history

Commits on Oct 22, 2024

  1. Use importlib.util instead of compile and exec to invoke Triton

    @voltjia
    voltjia committed Oct 22, 2024
    Configuration menu
    Copy the full SHA
    cdba870 View commit details
    Browse the repository at this point in the history

  2. Add support for calling Triton code from NineToothed

    @voltjia
    voltjia committed Oct 22, 2024
    Configuration menu
    Copy the full SHA
    45ed047 View commit details
    Browse the repository at this point in the history

  3. Treat a 0-dimensional tensor as a scalar

    @voltjia
    voltjia committed Oct 22, 2024
    Configuration menu
    Copy the full SHA
    99998d7 View commit details
    Browse the repository at this point in the history

  4. Prefix tensor names with _ninetoothed_

    @voltjia
    voltjia committed Oct 22, 2024
    Configuration menu
    Copy the full SHA
    5a20e4b View commit details
    Browse the repository at this point in the history

  5. Add test for addmm

    @voltjia
    voltjia committed Oct 22, 2024
    Configuration menu
    Copy the full SHA
    7edeccf View commit details
    Browse the repository at this point in the history

  6. Increment the version number from 0.5.0 to 0.6.0

    @voltjia
    voltjia committed Oct 22, 2024
    Configuration menu
    Copy the full SHA
    8fe5586 View commit details
    Browse the repository at this point in the history