Skip to content
This repository has been archived by the owner on Aug 7, 2024. It is now read-only.

add option for using fused kernel #227

Draft
wants to merge 11 commits into
base: main
Choose a base branch
from
Draft

add option for using fused kernel #227

wants to merge 11 commits into from

Commits on Mar 5, 2024

  1. add option for using fused kernel

    @drisspg
    drisspg committed Mar 5, 2024
    Configuration menu
    Copy the full SHA
    88f7a87 View commit details
    Browse the repository at this point in the history

  2. more support

    @drisspg
    drisspg committed Mar 5, 2024
    Configuration menu
    Copy the full SHA
    fe3485b View commit details
    Browse the repository at this point in the history

  3. add two more kernels

    @drisspg
    drisspg committed Mar 5, 2024
    Configuration menu
    Copy the full SHA
    d61b1f8 View commit details
    Browse the repository at this point in the history

  4. change the test shapes

    @drisspg
    drisspg committed Mar 5, 2024
    Configuration menu
    Copy the full SHA
    548dceb View commit details
    Browse the repository at this point in the history

  5. split reduction

    @drisspg
    drisspg committed Mar 5, 2024
    Configuration menu
    Copy the full SHA
    e6a1e3c View commit details
    Browse the repository at this point in the history

  6. use bigger reduce block

    @drisspg
    drisspg committed Mar 5, 2024
    Configuration menu
    Copy the full SHA
    da2ab56 View commit details
    Browse the repository at this point in the history

  7. need to handle braodcasted grads

    @drisspg
    drisspg committed Mar 5, 2024
    Configuration menu
    Copy the full SHA
    4526eb8 View commit details
    Browse the repository at this point in the history

  8. this is failing in backward for some reason with device context failure

    @drisspg
    drisspg committed Mar 5, 2024
    Configuration menu
    Copy the full SHA
    df940ae View commit details
    Browse the repository at this point in the history

  9. my abs_max is busted

    @drisspg
    drisspg committed Mar 5, 2024
    Configuration menu
    Copy the full SHA
    7c0778a View commit details
    Browse the repository at this point in the history

  10. fix kernel args

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

  11. cast to fp32 in amax

    @drisspg
    drisspg committed Mar 5, 2024
    Configuration menu
    Copy the full SHA
    110ec4b View commit details
    Browse the repository at this point in the history