Kernels

Commit History

  1. add build

    56776ac
  2. add readme with precommit hooks and applied pre commit to all files

    f517c97

  1. add pre-commit

    a73a0c0
  2. optimize poly norm kernel

    f8a7e6f

  1. add poly_norm_naive.cui for temp test

    18ec195

  1. torch 2.8 support (#1)⁠

    Verified signature
    43629b7

  1. feat: add cuda build

    cf68df1

  1. feat: support reset_parameters()

    605f22e
  2. feat(rms-norm): Impl fused RMSNorm

    f3b99fb

  1. feat(poly-norm): add perf test

    d14fd4d

  1. fix(poly-norm): calc param grad explicitly

    704692b
  2. fix(poly-norm): fix bug in reduce sum

    883cc1c
  3. refactor(poly-norm): fix indentation

    32c2bde

  1. refactor(poly-norm): use const for immutable args

    e85ecc9
  2. chore(poly-norm): remove unnecessary file

    552d415
  3. chore: use latest build image and misc

    f5a7d38

  1. chore(poly-norm): update README and build artifacts

    f72121c
  2. feat(poly-norm): add default value for eps argument

    afd2a56
  3. chore(poly-norm): add ROCm build artifacts

    4b70498
  4. feat(poly-norm): Add PolyNorm

    44e9845
  5. initial commit

    Verified signature
    7a7d761