ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-19 22:39:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Grouped conv fwd with direct load (#3082)

Browse Source 
* Grouped conv fwd with direct load

* fix

* fix

* Add IsSupported check

* Fix

* fix inductor
This commit is contained in:
Bartłomiej Kocot
2025-10-29 09:54:42 +01:00
committed by GitHub
parent 3052d7c9e6
commit 66bae4306c
27 changed files with 2165 additions and 285 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/ck4inductor/grouped_conv_fwd/op.py
View File

@@ -66,6 +66,8 @@ class CKGroupedConvFwdOp:
a_compute_dtype: Optional[str] = None
b_compute_dtype: Optional[str] = None
direct_load: Optional[bool] = None
def name(self):
# cpp alias for template instance
return (