qazalandGitHub 1cf8f2f68c llama: inplace amax update (#17064) ...

* llama: inplace amax update

* remove amax_out return

* work

* fit

* work

* work

* keep

* diff cleanup

2026-07-20 15:05:41 +09:00

..

__init__.py

llama: inplace amax update (#17064)

2026-07-20 15:05:41 +09:00

cast_amax_bwd_w13.cpp

llama: replace two stage amax with atomics (#17063)

2026-07-17 19:27:10 +09:00

cast_amax_fwd_w13.cpp

llama: replace two stage amax with atomics (#17063)

2026-07-17 19:27:10 +09:00