Ville Pietilä
faf07cc3ab
Code clean-up.
2025-10-07 10:41:18 +00:00
Ville Pietilä
9519405b4a
Merge remote-tracking branch 'origin/develop' into vpietila/merge-multiple-conv-groups-into-single-wg-in-ck-tile
2025-10-07 08:05:32 +00:00
Ville Pietilä
9554344c94
Fix tensor descriptors for merged conv groups when K > 1.
2025-10-06 12:57:47 +00:00
Ville Pietilä
24fe5e4f80
Fix bugs in merged conv groups tensor descriptors.
2025-10-06 10:22:23 +00:00
Ville Pietilä
48d22d2b9b
Remove the obsolete template parameters.
2025-10-03 14:36:48 +00:00
Ville Pietilä
99fe3df99a
Fix tensor descriptors.
2025-10-03 14:23:04 +00:00
Ville Pietilä
9510171377
WIP: Put back the generic tensor descriptors for convolutions.
2025-10-02 15:06:30 +00:00
Ville Pietilä
db835e065c
Make MPerGroup and NPerGroup template parameters.
2025-09-30 07:14:28 +00:00
Ville Pietilä
558054eadb
WIP: Simplify conv to gemm transformations and handle K > 1 and C > 1 cases.
2025-09-26 13:38:24 +00:00
Ville Pietilä
8babf7195a
Fix strides in 1D conv to gemm transformation.
2025-09-26 09:38:11 +00:00
Ville Pietilä
b864c077ed
Code clean-up for bwd tensor transformations.
2025-09-25 15:09:08 +00:00
Ville Pietilä
29e3112b9b
Epilogue fixes.
2025-09-22 15:38:02 +00:00
jakpiase
624c46866e
[CK_TILE] Add conv bwd weight two stage support ( #2855 )
...
* resolved conflicts
* add conv bwd weight twostage
* fix one file
* fixes after review
* fixes
* fixes
* Fix
---------
Co-authored-by: Bartlomiej Kocot <barkocot@amd.com >
2025-09-22 15:31:25 +02:00
JH-Leon-KIM-AMD
804065a36b
[CK Tile] Grouped conv fwd splitn support ( #2776 )
...
## What's New
Add Split-N support for grouped convolution forward to handle tensors >2GB by splitting the batch dimension.
## Bug Fix
Fixed 32-bit integer overflow that caused crashes with 6+ splits:
- Use `long_index_t` for batch offset calculations
- Remove redundant GemmM initialization in constructors
## How It Works
- Automatically splits batch dimension when tensor exceeds 2GB
- Uses grid.z dimension for parallel processing of splits
- Each split processes a subset of batches independently
## Testing
Verified with tile_example_grouped_conv_fwd:
- n=3000 (6 splits) ✓
- n=3500 (7 splits) ✓
- n=10480 (40 splits) ✓
2025-09-16 16:56:11 +03:00
Ville Pietilä
0d5c1b9638
WIP: Merged conv groups epilogue.
2025-09-11 15:24:36 +00:00
Ville Pietilä
d9f0a9cdd0
Fully working conv group merging for TransformConvBwdWeightToGemm.
2025-09-09 09:58:43 +00:00
Ville Pietilä
8845b23254
WIP: Tensor transformations.
2025-09-08 15:41:54 +00:00
Ville Pietilä
61b3c96273
Add number of groups to merge to ck tile grouped gemm example.
2025-09-04 14:24:23 +00:00
Bartłomiej Kocot
4212bbc170
[CK Tile] Grouped convolution backward data ( #2652 )
...
* base working version for single groupped conv bwd data
* Fix 2d descriptor
* fix groups
* Add 3d support
* fixes
* fixes
* fixes
---------
Co-authored-by: Jakub Piasecki <jakpia21@gmail.com >
2025-08-20 05:29:57 -07:00
Illia Silin
504b101da3
upgrade from clang-format-12 to clang-format-18 ( #2568 )
...
* upgrade to clang-format-18
* update to clang-format-18 in pre-commit-config
2025-07-28 11:34:07 -07:00
jakpiase
6681593864
[CK_TILE] Grouped Convolution Backward Weight Kernel ( #2357 )
...
* [CK TILE] Grouped Convolution Forward Kernel
* custom vector size
* fixes
* refactor
* resolved conflicts
* rebase fixes
* fixes
* tmp
* add working support for splitk
* minor fix
* fixes
* fixes
* minor fix
* small fix
* Split K and preprocessing fixes
---------
Co-authored-by: Bartlomiej Kocot <barkocot@amd.com >
2025-07-24 10:41:35 +02:00
Bartłomiej Kocot
cebdee4d9e
[CK TILE] Grouped Convolution Forward Kernel ( #2188 )
...
* [CK TILE] Grouped Convolution Forward Kernel
* custom vector size
* fixes
* refactor
* rebase fixes
* fixes
* fixes
2025-06-20 15:44:36 -07:00
