[CK] Allow tensors larger than 2GB in grouped conv bwd weight (#3169)
* Take split_k into account when checking 2GB tensor limit. * Revert "Take split_k into account when checking 2GB tensor limit." This reverts commit adf35c91be5b25bd753f12a6f1252146a9a7aca1. * Optimize grouped conv bwd wei split_k off calc (cherry picked from commit 6f61dd56c5d45409826e660175accb51ace24bcc) * Update gridwise_gemm_xdl_cshuffle_conv_v3.hpp (cherry picked from commit b33877c10f7257b58e952659e35ba4ade61da58a) * 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 commit 3883fe693571fcf61e7860b2973523c5bd52c981. This reverts commit 5311ec608d2fd5fa7835b213f8dc714ca68ce8d2. * Reapply "arch-guarded atomic_add implementations for gfx11" This reverts commit 1972adeddc0bf4abda4e90c315000d5bb9da826c. * Only remove float4 atomic_add * Refactor to single flag * Consolidate template parameters * Consolidate flag in transformers --------- Co-authored-by: Bartlomiej Kocot <barkocot@amd.com>
J
Johannes Graner committed
ee2c35b92db5ef4c4703935d203e9612e6b5f573
Parent: bc497be
Committed by GitHub <noreply@github.com>
on 1/8/2026, 7:02:02 AM