[CK_BUILDER] Add bwd weight factories#3509

Merged

vpietila-amd merged 72 commits intodevelopfrom

vpietila/ckb-bwd-weight-factories

Jan 13, 2026

Commits on Dec 18, 2025

Add placeholder test.
Ville Pietilä
committed

Commits on Dec 19, 2025

Merge remote-tracking branch 'origin/develop' into vpietila/ckb-bwd-weight-factories
Ville Pietilä
committed
Initial conv bwd weight factory.
Ville Pietilä
committed
Conv builder test refactoring.
Ville Pietilä
committed
Add missing pieces to bwd weight factory.
Ville Pietilä
committed

Commits on Dec 22, 2025

Improve compile time erros message when no matching factory is found.
Ville Pietilä
committed
Use amcro to ensure automatic macthing between concepts are their string representations.
Ville Pietilä
committed
Improve compile time diagnostics.
Ville Pietilä
committed
Small improvements.
Ville Pietilä
committed
Improve missing member/wrong type compile-time errors.
Ville Pietilä
committed
Improve compile time diagnostics.
Ville Pietilä
committed
Concept bug fixes.
Ville Pietilä
committed
Remove debug assert.
Ville Pietilä
committed
Update algorithm signature diagnostics.
Ville Pietilä
committed
Factory bug fixes.
Ville Pietilä
committed
First functional version of bwd weight conv factory.
Ville Pietilä
committed

Commits on Dec 23, 2025

Refactor handing of GEMM-K batch template parameter in conv bwd weight factory.
Ville Pietilä
committed
Concept improvements.
Ville Pietilä
committed

Commits on Dec 29, 2025

Improve concept diagnostics.
Ville Pietilä
committed
Introduve a common size type for concepts.
Ville Pietilä
committed
Update compiletime diagnostics to use the size type.
Ville Pietilä
committed
Update conv specialization enum.
Ville Pietilä
committed
Fix fwd conv builder tests.
Ville Pietilä
committed
Fix smoke tests.
Ville Pietilä
committed
Separate bwd weigth and bwd data tests into separate targets.
Ville Pietilä
committed
Clean-up CK Tile builder tests.
Ville Pietilä
committed
Add bwd weight XDL CShuffle V3 factory.
Ville Pietilä
committed
Build conv bwd weigth v3 instances successfully.
Ville Pietilä
committed
Add instance traits for DeviceGroupedConvBwdWeight_Xdl_CShuffleV3.
Ville Pietilä
committed
Test fix.
Ville Pietilä
committed

Commits on Dec 30, 2025

Add instance traits for bwd weight algorithms.
Ville Pietilä
committed
Add unit tests for instance strings.
Ville Pietilä
committed
Build new instance traits unit tests but exclude WMMA for now.
Ville Pietilä
committed

Commits on Dec 31, 2025

Added factory for DeviceGroupedConvBwdWeightTwoStage_Xdl_CShuffle.
Ville Pietilä
committed
Conv bwd weight DL factory.
Ville Pietilä
committed
Final implementation for bwd weight DL factory.
Ville Pietilä
committed
Add test for creating DeviceGroupedConvBwdWeightMultipleD_Xdl_CShuffle instance.
Ville Pietilä
committed
Add factory for DeviceGroupedConvBwdWeightMultipleD_Xdl_CShuffle
Ville Pietilä
committed

Commits on Jan 2, 2026

Merge remote-tracking branch 'origin/develop' into vpietila/ckb-bwd-weight-factories
Ville Pietilä
committed
Treat ref algorithm the same way as real algorithms in the dispatcher.
Ville Pietilä
committed
Refactor large tensor support and WMMA configuration.
Ville Pietilä
committed
Add factory and tests for DeviceGroupedConvBwdWeight_Wmma_CShuffleV3.
Ville Pietilä
committed
Update Readme.
Ville Pietilä
committed
Fix WMMA bwd weight tests.
Ville Pietilä
committed
Added factory and tests for DeviceGroupedConvBwdWeightTwoStage_Wmma_CShuffleV3.
Ville Pietilä
committed
Factory and tests for DeviceGroupedConvBwdWeight_Wmma_CShuffle.
Ville Pietilä
committed
Dispatching for DeviceGroupedConvBwdWeightMultipleD_Wmma_CShuffle.
Ville Pietilä
committed
Add factory for DeviceGroupedConvBwdWeightMultipleD_Wmma_CShuffleV3
Ville Pietilä
committed

Commits on Jan 5, 2026

Fix DeviceGroupedConvBwdWeightMultipleD_Wmma_CShuffleV3 factory and compute types for input and output tensor in bwd weigth convs.
Ville Pietilä
committed
Fix fwd factories after refactoring.
Ville Pietilä
committed
clang-format
Ville Pietilä
committed
Move compile-time diagnostics to a separate branch.
Ville Pietilä
committed
Merge branch 'develop' into vpietila/ckb-bwd-weight-factories
vpietila-amd
authored

Commits on Jan 7, 2026

Fix ref algorithm dispatching.
Ville Pietilä
committed
Fix smoke tests.
Ville Pietilä
committed
clang-format
Ville Pietilä
committed
Fix factory for regular WMMA conv bwd weight.
Ville Pietilä
committed
Merge branch 'develop' into vpietila/ckb-bwd-weight-factories
vpietila-amd
authored

Commits on Jan 8, 2026

Merge remote-tracking branch 'origin/develop' into vpietila/ckb-bwd-weight-factories
Ville Pietilä
committed
Clarify builder Readme.
Ville Pietilä
committed
Remove obsolete test file.
Ville Pietilä
committed
Fix test after merge.
Ville Pietilä
committed
clang-format
Ville Pietilä
committed

Commits on Jan 12, 2026

Merge branch 'develop' into vpietila/ckb-bwd-weight-factories
vpietila-amd
authored
Remove the C++26 extensions.
Ville Pietilä
committed

Commits on Jan 13, 2026

Unify conv elementwise ops and layout definitions for fwd and bwd directions.
Ville Pietilä
committed
Remove old layout and elementwise ops.
Ville Pietilä
committed
Unify handling of conv tensor types between fwd and bwd directions.
Ville Pietilä
committed
Unify block transfer for fwd and bwd directions. Rename ThreadSliceDim to ThreadClusterRank.
Ville Pietilä
committed
Make BlockTransferDescriptor concept parametrized. Introduce a common TileTransferParameters concept for conv algorithms.
Ville Pietilä
committed
clang-format
Ville Pietilä
committed
Merge remote-tracking branch 'origin/develop' into vpietila/ckb-bwd-weight-factories
Ville Pietilä
committed