SIGN IN SIGN UP

[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