composable_kernel/test/ck_tile/pooling/CMakeLists.txt at abc0a0b77f99598c8ebee290b3fdbcd3ec0f12c3 - composable_kernel - Public git mirror

ROCm/composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-18 20:09:25 +00:00

Files

Yashvardhan Agarwal bc9e9df38f [CK_TILE] Pooling FWD (Lwpck 3683) (#2956 )

* Pooling 2D/3D with refernce

* Tests & cleanup

- added test for ppoling
- cleanup
- removed 2d example

* Comment resolution

- README added
- example target name rectified
- appropriate arg description and comments added

* clang-format

* appropriate blocksize calc

* modifications for future indexing addition

- instead of transforming views we now transform the descriptors, so
that the same descriptor can be re-used for index tensor in the future

* some basic fixes

* comment resolutions

* comment resolutions

---------

Co-authored-by: Illia Silin <98187287+illsilin@users.noreply.github.com>

[ROCm/composable_kernel commit: 7b6451b68e]

2025-10-09 16:13:26 +02:00

4 lines

115 B

CMake

Raw Blame History

	`if(GPU_TARGETS MATCHES "gfx9\|gfx11\|gfx12")`
	`add_gtest_executable(test_ck_tile_pooling test_pooling.cpp)`
	`endif()`