ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-20 06:49:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update tile_engine/ops/pooling/pool_instance_builder.py

Browse Source 
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
This commit is contained in:
Thomas Ning
2026-02-19 11:17:53 -08:00
committed by GitHub
parent 39ad7898fa
commit 9e97c2dc40
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tile_engine/ops/pooling/pool_instance_builder.py
View File

@@ -35,7 +35,7 @@ def get_reduce_op_string(reduce_op):
"max": "ck_tile::ReduceOp::Max",
"min": "ck_tile::ReduceOp::Min",
"add": "ck_tile::ReduceOp::Add",
"avg": "ck_tile::ReduceOp::Add", # Average uses Add and divides later
"avg": "ck_tile::ReduceOp::Add", # Average pooling uses Add here; division by the element count is performed elsewhere in the pooling kernel/reference implementation.
}
return reduce_op_map.get(reduce_op.lower(), "ck_tile::ReduceOp::Max")