mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-06-28 18:56:59 +00:00
[rocm-libraries] ROCm/rocm-libraries#7734 (commit 03ffb9d)
[CK] Grouped Convolution Global Load/Store instances ## Motivation Support global load and store in grouped convolutions using instance factory. ## Technical Details - add new instances for each direction - add new tests for large cases ## Test Plan New test for large cases ## Test Result pending ## Submission Checklist - [x] Look over the contributing guidelines at https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests. AICK-1255
This commit is contained in:
committed by
assistant-librarian[bot]
parent
2c363870d9
commit
28f2966762
@@ -68,6 +68,7 @@ class CKGroupedConvFwdOp:
|
||||
|
||||
direct_load: Optional[bool] = None
|
||||
num_groups_to_merge: Optional[int] = None
|
||||
large_tensor: Optional[bool] = None
|
||||
|
||||
def name(self):
|
||||
# cpp alias for template instance
|
||||
|
||||
Reference in New Issue
Block a user