Kernels
activation / torch-ext /torch_binding.cpp

Commit History

refactor(activation): change fused_add_rms_norm and fused_add_rms_norm_backward to out-place operations
7e4334d

wyldecat commited on

feat: make rms_norm as out-place
9d0a235

wyldecat commited on

Fix fused add rms norm (#4)
a1e5ca8
unverified

TaehyunKim TaehyunKimMotif commited on

feat(rms-norm): Impl fused RMSNorm
f3b99fb

iamwyldecat commited on

chore: use latest build image and misc
f5a7d38

iamwyldecat commited on

feat(poly-norm): Add PolyNorm
44e9845

iamwyldecat commited on