- Add .t() transpose to all weight matrix multiplications - Add .contiguous() after transpose to fix non-contiguous errors - Fix causal mask using additive masking instead of where_cond - Fix mask dtype compatibility (F32 instead of U8) All 8 quantized_attention tests now passing.
7.3 KiB
7.3 KiB