Kernels

Commit History

add readme with precommit hooks and applied pre commit to all files
f517c97

TaehyunKimMotif commited on

add pre-commit
a73a0c0

ca1207 commited on

optimize poly norm kernel
f8a7e6f

ca1207 commited on

add poly_norm_naive.cui for temp test
18ec195

ca1207 commited on

torch 2.8 support (#1)
43629b7
verified

iamwyldecat commited on

feat: add cuda build
cf68df1

iamwyldecat commited on

feat: support reset_parameters()
605f22e

iamwyldecat commited on

feat(rms-norm): Impl fused RMSNorm
f3b99fb

iamwyldecat commited on

feat(poly-norm): add perf test
d14fd4d

iamwyldecat commited on

fix(poly-norm): calc param grad explicitly
704692b

iamwyldecat commited on

fix(poly-norm): fix bug in reduce sum
883cc1c

iamwyldecat commited on

refactor(poly-norm): fix indentation
32c2bde

iamwyldecat commited on

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

iamwyldecat commited on

chore(poly-norm): remove unnecessary file
552d415

iamwyldecat commited on

chore: use latest build image and misc
f5a7d38

iamwyldecat commited on

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

iamwyldecat commited on

feat(poly-norm): add default value for eps argument
afd2a56

iamwyldecat commited on

chore(poly-norm): add ROCm build artifacts
4b70498

iamwyldecat commited on

feat(poly-norm): Add PolyNorm
44e9845

iamwyldecat commited on

initial commit
7a7d761
verified

iamwyldecat commited on