The FP8 backbone mode added each linear's bias by downloading, tiling and re-uploading it per call, which cost more than the FP8 GEMM saved. A gpu_add_cols_broadcast op (CUDA kernel; host loop on the Metal tensor backend) adds a cols-wide bias to every row on the device, and each linear keeps its bias resident after the first upload. Co-Authored-By: Claude Fable 5.1 <noreply@anthropic.com> |
||
|---|---|---|
| .. | ||
| kernels | ||
| src | ||
| tests/data | ||
| build.rs | ||
| Cargo.toml | ||