DEV
GPU & INFERENCE
+40 758 652 786 CONTACT@MOMODEV.NET

Things we built
without a client.

Kernel work and benchmarks we run on our own time, with the same tools and the same rigour we bring to your cluster.

PROJECTS

FlashAttention-3, variable-length.

Profiling and optimising variable-length attention kernels on NVIDIA Hopper, aiming for an upstream contribution to FlashAttention-3.

IN PROGRESSCUDAFLASHATTENTIONHOPPERNSIGHT
1.3×VARLEN FORWARD VS BASELINE, H100

Reduction, softmax and GEMM, by hand.

Hand-optimised kernels, profiled with Nsight Compute and checked against the roofline at every step. Every version is benchmarked against the last.

IN PROGRESSCUDANSIGHT COMPUTEROOFLINEGEMM
91%OF cuBLAS, FP16 GEMM ON H100

BENCHMARK · H100

REDUCTION 3.1 TB/s · 92% OF PEAK BW
SOFTMAX 2.9 TB/s · 87% OF PEAK BW
GEMM, FP16 91% OF cuBLAS
FA3 VARLEN FWD 1.3× VS BASELINE