ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-03 21:21:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add conv fwd/bwd data scale instances, extend bilinear instances (#1178)

Browse Source 
* Add conv fwd/bwd data scale instances

* Fix cmake client example file

---------

Co-authored-by: Adam Osewski <19374865+aosewski@users.noreply.github.com>
This commit is contained in:
Bartłomiej Kocot
2024-03-13 23:09:08 +01:00
committed by GitHub
parent 12441af014
commit 285251768e
19 changed files with 1637 additions and 89 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

@@ -310,6 +310,12 @@ struct Scale
y = scale_ * x;
};
template <>
__host__ __device__ void operator()<int8_t, int8_t>(int8_t& y, const int8_t& x) const
{
y = ck::type_convert<int8_t>(scale_ * ck::type_convert<float>(x));
};
float scale_;
};