[Layers] Enable AMX FP16 of FlashAttn #459

Closed abenmao closed 1 week ago