aledudek
|
bacec21478
|
Extend pool3d fwd avg, max operations by f8_t, int8_t types (#1483)
* Extend pool3d fwd avg, max operations by f8_t, int8_t types
* Pack MaxPool3dFwd params together
* Fix MaxPool3dFwd AVG instances
* Decrease verification precision for bf16
* Adjust tests + review changes
* Adjust threshold for F8
* Adjusted compute types for MAX op instances
* Fix ComputeDataType mismatch in tests and profiler for AVG
* Fix naming from max_pool3d_fwd to pool3d_fwd
* Adjust CMakeLists
---------
Co-authored-by: Adam Osewski <19374865+aosewski@users.noreply.github.com>
[ROCm/composable_kernel commit: a793afc961]
|
2024-09-17 15:57:10 +02:00 |
|