Commit 698f3a3e authored by Chao Liu's avatar Chao Liu
Browse files

update example

parent 96f3935c
add_example_executable(example_gemm_xdl_alpha_beta gemm_xdl_alpha_beta.cpp) add_example_executable(example_gemm_alpha_beta_xdl_fp16 gemm_alpha_beta_xdl_fp16.cpp)
...@@ -28,7 +28,7 @@ struct AlphaBetaAdd ...@@ -28,7 +28,7 @@ struct AlphaBetaAdd
__host__ __device__ constexpr void operator()<ck::half_t, float, ck::half_t>( __host__ __device__ constexpr void operator()<ck::half_t, float, ck::half_t>(
ck::half_t& e, const float& c, const ck::half_t& d) const ck::half_t& e, const float& c, const ck::half_t& d) const
{ {
e = ck::type_convert<ck::half_t>(alpha_ * c; beta_ * ck::type_convert<float>(d)); e = ck::type_convert<ck::half_t>(alpha_ * c + beta_ * ck::type_convert<float>(d));
}; };
float alpha_; float alpha_;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment