3 ms·
This is super cool! Especially matrix mult getting similar or better perf than cuBLAS! If anyone is interested on other kernels like swiglu, geglu, RMS layerno
by danielhanchen 2y ago
This is super cool! Especially matrix mult getting similar or better perf than cuBLAS! If anyone is interested on other kernels like swiglu, geglu, RMS layernorm, I coded some at https://github.com/unslothai/unsloth/tree/main/unsloth/kernels https://github.com/unslothai/unsloth/tree/main/unsloth/kerne...
- zaptrem 2y agoNeat! Are these compatible with torch compile?