- wrap one-thread map + shared broadcast + early exit - both scalar and MMA prefill kernels use the shared helper