"...composable_kernel_rocm.git" did not exist on "bdcd037428ac356e5b77271b7b6669c5c2d9548a"
Commit 94d5f723 authored by rocking's avatar rocking
Browse files

Fix static_cast order

parent 3d091db2
...@@ -169,11 +169,11 @@ void host_gemm_layernorm(Tensor<LayerNormOutDataType>& out_m_n, ...@@ -169,11 +169,11 @@ void host_gemm_layernorm(Tensor<LayerNormOutDataType>& out_m_n,
AccDataType acc = AccDataType acc =
static_cast<AccDataType>(c_m_n(m, n)) + static_cast<AccDataType>(bias_n(n)); static_cast<AccDataType>(c_m_n(m, n)) + static_cast<AccDataType>(bias_n(n));
AccDataType c1 = c1_m_n(m, n); AccDataType c1 = static_cast<AccDataType>(c1_m_n(m, n));
c_element_op(acc, acc); c_element_op(acc, acc);
c1_element_op(c1, c1); c1_element_op(c1, c1);
acc += static_cast<AccDataType>(c1); acc += c1;
c_m_n(m, n) = static_cast<CDataType>(acc); c_m_n(m, n) = static_cast<CDataType>(acc);
} }
......
...@@ -188,10 +188,10 @@ void profile_gemm_bias_add_reduce_impl(int do_verification, ...@@ -188,10 +188,10 @@ void profile_gemm_bias_add_reduce_impl(int do_verification,
ReduceAccDataType acc = static_cast<ReduceAccDataType>(c_m_n_host_result(m, n)) + ReduceAccDataType acc = static_cast<ReduceAccDataType>(c_m_n_host_result(m, n)) +
static_cast<ReduceAccDataType>(bias_n(n)); static_cast<ReduceAccDataType>(bias_n(n));
ReduceAccDataType c1 = c1_m_n(m, n); ReduceAccDataType c1 = static_cast<ReduceAccDataType>(c1_m_n(m, n));
c_element_op(acc, acc); c_element_op(acc, acc);
c1_element_op(c1, c1); c1_element_op(c1, c1);
acc += static_cast<ReduceAccDataType>(c1); acc += c1;
c_m_n_host_result(m, n) = static_cast<CDataType>(acc); c_m_n_host_result(m, n) = static_cast<CDataType>(acc);
} }
......
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