Commits · 2a4c23161ad252a8f5ce454eccf48a0e77e6e968 · gaoqiong / composable_kernel_ROCM

12 Sep, 2024 1 commit

Pool2d max/avg kernel in the BWD version (#1494) · 448c0f56

Mateusz Ozga authored Sep 12, 2024

* Add pool2d instance BWD AVG

* Add pool2d instance BWD MAX

* Fix: avg review

* Fix review: part2

* Fix - enable test when type is compiled

* Fix review part3

448c0f56