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

[WIP] Add custom triton kernels #84

Merged
merged 24 commits into from
Jul 29, 2024
Merged

[WIP] Add custom triton kernels #84

merged 24 commits into from
Jul 29, 2024

Commits on Jul 22, 2024

  1. add custom triton kernels

    Mehmet Cagri Kaymak committed Jul 22, 2024
    Configuration menu
    Copy the full SHA
    1ed39dd View commit details
    Browse the repository at this point in the history

  2. add CPU compatibility

    @lubbersnick
    lubbersnick committed Jul 22, 2024
    Configuration menu
    Copy the full SHA
    b628310 View commit details
    Browse the repository at this point in the history

  3. move call site for CPU version

    @lubbersnick
    lubbersnick committed Jul 22, 2024
    Configuration menu
    Copy the full SHA
    727c502 View commit details
    Browse the repository at this point in the history

  4. call the right functions

    @lubbersnick
    lubbersnick committed Jul 22, 2024
    Configuration menu
    Copy the full SHA
    934413f View commit details
    Browse the repository at this point in the history

  5. fix GPU memory limits for tests

    @lubbersnick
    lubbersnick committed Jul 22, 2024
    Configuration menu
    Copy the full SHA
    28619e0 View commit details
    Browse the repository at this point in the history

  6. set the seed for repro. and change triton loading logic

    Mehmet Cagri Kaymak committed Jul 22, 2024
    Configuration menu
    Copy the full SHA
    6931aa2 View commit details
    Browse the repository at this point in the history

Commits on Jul 23, 2024

  1. update compare_against

    Nicholas Lubbers committed Jul 23, 2024
    Configuration menu
    Copy the full SHA
    e2b8901 View commit details
    Browse the repository at this point in the history

  2. update tester

    @lubbersnick
    lubbersnick committed Jul 23, 2024
    Configuration menu
    Copy the full SHA
    c88086b View commit details
    Browse the repository at this point in the history

Commits on Jul 24, 2024

  1. small tweaks to testing arguments

    Nicholas Lubbers committed Jul 24, 2024
    Configuration menu
    Copy the full SHA
    298b258 View commit details
    Browse the repository at this point in the history

  2. integrate triton kernels to options, apply formatter

    @lubbersnick
    lubbersnick committed Jul 24, 2024
    Configuration menu
    Copy the full SHA
    5df6edd View commit details
    Browse the repository at this point in the history

  3. adjust configparser

    Nicholas Lubbers committed Jul 24, 2024
    Configuration menu
    Copy the full SHA
    ecf706b View commit details
    Browse the repository at this point in the history

Commits on Jul 25, 2024

  1. try refactor triton code (cosmetic)

    @lubbersnick
    lubbersnick committed Jul 25, 2024
    Configuration menu
    Copy the full SHA
    4f9b389 View commit details
    Browse the repository at this point in the history

  2. formatter

    @lubbersnick
    lubbersnick committed Jul 25, 2024
    Configuration menu
    Copy the full SHA
    ab9d47a View commit details
    Browse the repository at this point in the history

  3. add ultra-size test

    Nicholas Lubbers committed Jul 25, 2024
    Configuration menu
    Copy the full SHA
    dec9fdd View commit details
    Browse the repository at this point in the history

  4. remove explicit numba dependency from custom kernel tests

    @lubbersnick
    lubbersnick committed Jul 25, 2024
    Configuration menu
    Copy the full SHA
    1ecd758 View commit details
    Browse the repository at this point in the history

  5. update triton to use numba on CPU

    @lubbersnick
    lubbersnick committed Jul 25, 2024
    Configuration menu
    Copy the full SHA
    4aab176 View commit details
    Browse the repository at this point in the history

  6. more formatting and name changes

    @lubbersnick
    lubbersnick committed Jul 25, 2024
    Configuration menu
    Copy the full SHA
    fc7cb0a View commit details
    Browse the repository at this point in the history

  7. fix lack of forward correctness checks!

    @lubbersnick
    lubbersnick committed Jul 25, 2024
    Configuration menu
    Copy the full SHA
    d1c4e37 View commit details
    Browse the repository at this point in the history

  8. actually do what the last commit says

    @lubbersnick
    lubbersnick committed Jul 25, 2024
    Configuration menu
    Copy the full SHA
    bd5c7df View commit details
    Browse the repository at this point in the history

  9. update for todos

    @lubbersnick
    lubbersnick committed Jul 25, 2024
    Configuration menu
    Copy the full SHA
    286b379 View commit details
    Browse the repository at this point in the history

Commits on Jul 26, 2024

  1. split feat. and sense vectors to chunks to lower register pressure, a… 

    …dd autotune
    Mehmet Cagri Kaymak committed Jul 26, 2024
    Configuration menu
    Copy the full SHA
    4265297 View commit details
    Browse the repository at this point in the history

Commits on Jul 27, 2024

  1. raise better errors in old custom kernels. partially update documenta… 

    …tion
    @lubbersnick
    lubbersnick committed Jul 27, 2024
    Configuration menu
    Copy the full SHA
    2a7fa6d View commit details
    Browse the repository at this point in the history

Commits on Jul 29, 2024

  1. update documentation for custom kernels

    @lubbersnick
    lubbersnick committed Jul 29, 2024
    Configuration menu
    Copy the full SHA
    fa5845b View commit details
    Browse the repository at this point in the history

  2. update requirements/docs

    @lubbersnick
    lubbersnick committed Jul 29, 2024
    Configuration menu
    Copy the full SHA
    aa2c833 View commit details
    Browse the repository at this point in the history