mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-14 02:02:46 +00:00
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:
@@ -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