mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-12 17:26:00 +00:00
[rocm-libraries] ROCm/rocm-libraries#6434 (commit 87aae5c)
Fix ck4inductor conv instance parsing for NumGroupsToMerge parameter (#6434) ## Summary - Add `num_groups_to_merge` field to `CKGroupedConvFwdOp` dataclass to match the new (#4273) `NumGroupsToMerge` template parameter added to `DeviceGroupedConvFwdMultipleABD_Xdl_CShuffle_V3` - Enable inductor tests by default in Jenkins CI ## Test plan - [x] Built wheel without patch: `test_gen_conv_instances` fails with `TypeError: takes from 47 to 50 positional arguments but 51 were given` - [x] Built wheel with patch: `test_gen_conv_instances` passes
This commit is contained in:
committed by
assistant-librarian[bot]
parent
cbfb3e242e
commit
1ca05a105a
@@ -67,6 +67,7 @@ class CKGroupedConvFwdOp:
|
||||
b_compute_dtype: Optional[str] = None
|
||||
|
||||
direct_load: Optional[bool] = None
|
||||
num_groups_to_merge: Optional[int] = None
|
||||
|
||||
def name(self):
|
||||
# cpp alias for template instance
|
||||
|
||||
Reference in New Issue
Block a user