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

Add transpose scale amax example (#1547)

Browse Source 
* Add transpose scale amax example

* fixes

* Tune reduce instance
This commit is contained in:
Bartłomiej Kocot
2024-10-14 17:39:38 +02:00
committed by GitHub
parent 35c1777d59
commit f21cda2536
4 changed files with 258 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
include/ck/tensor_operation/gpu/element/unary_element_wise_operation.hpp
View File

@@ -419,6 +419,12 @@ struct UnaryAbs
y = ck::math::abs(x);
};
template <>
__host__ __device__ void operator()(f8_t& y, const f8_t& x) const
{
y = ck::type_convert<f8_t>(ck::math::abs(ck::type_convert<float>(x)));
};
};
struct UnarySqrt