mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-14 18:17:44 +00:00
9d6add54e5
* Take split_k into account when checking 2GB tensor limit. * Revert "Take split_k into account when checking 2GB tensor limit." This reverts commitadf35c91be. * Optimize grouped conv bwd wei split_k off calc (cherry picked from commit 2115642ee59050dabd81393c1b8f03b34adc05aa) * Update gridwise_gemm_xdl_cshuffle_conv_v3.hpp (cherry picked from commit 900d4d4b466f5730ae1189370d3c96267c35ea69) * Fix tensor descriptors and stride calculations * Don't miss half of the elements * Fix buffer size calculations * Disable hack if stride not divisible by k_batch * Clean up comments * Disallow hack in non-contiguous edge cases * Index -> Dim * Fix broken test * Refactor applicability checks into separate function * fix missed variable name * Fix variable name in info print * update V3 2GB check * No more regression, use templates instead * Code deduplication * Regression fix for cshuffle * arch-guarded atomic_add implementations for gfx11 * Similar for half(4|8)_t as well * Only use both offset hacks at the same time * Revert "arch-guarded atomic_add implementations for gfx11" This reverts commit3883fe6935. This reverts commit5311ec608d. * Reapply "arch-guarded atomic_add implementations for gfx11" This reverts commit1972adeddc. * Only remove float4 atomic_add * Refactor to single flag * Consolidate template parameters * Consolidate flag in transformers --------- Co-authored-by: Bartlomiej Kocot <barkocot@amd.com> [ROCm/composable_kernel commit:ee2c35b92d]