Commit e7a4548b authored by rocking's avatar rocking
Browse files

Fix typo

parent f1245c3b
...@@ -28,7 +28,7 @@ struct Bias ...@@ -28,7 +28,7 @@ struct Bias
{ {
__host__ __device__ constexpr void operator()(half_t& y_plus_bias, const half_t& y, const half_t& bias) const __host__ __device__ constexpr void operator()(half_t& y_plus_bias, const half_t& y, const half_t& bias) const
{ {
y_plus_bias = y_plus_bias = y + bias; y_plus_bias = y + bias;
} }
}; };
......
...@@ -176,9 +176,9 @@ int main(int argc, char* argv[]) ...@@ -176,9 +176,9 @@ int main(int argc, char* argv[])
Tensor<ADataType> a_m_k(f_host_tensor_descriptor(M, K, StrideA, ALayout{})); Tensor<ADataType> a_m_k(f_host_tensor_descriptor(M, K, StrideA, ALayout{}));
Tensor<BDataType> b_k_n(f_host_tensor_descriptor(K, N, StrideB, BLayout{})); Tensor<BDataType> b_k_n(f_host_tensor_descriptor(K, N, StrideB, BLayout{}));
Tensor<BDataType> c0_m_n(f_bias_host_tensor_descriptor(M, N)); Tensor<CDataType> c0_m_n(f_bias_host_tensor_descriptor(M, N));
Tensor<BDataType> c_m_n_host_result(f_host_tensor_descriptor(M, N, StrideC, CLayout{})); Tensor<CDataType> c_m_n_host_result(f_host_tensor_descriptor(M, N, StrideC, CLayout{}));
Tensor<BDataType> c_m_n_device_result(f_host_tensor_descriptor(M, N, StrideC, CLayout{})); Tensor<CDataType> c_m_n_device_result(f_host_tensor_descriptor(M, N, StrideC, CLayout{}));
std::cout << "a_m_k: " << a_m_k.mDesc << std::endl; std::cout << "a_m_k: " << a_m_k.mDesc << std::endl;
std::cout << "b_k_n: " << b_k_n.mDesc << std::endl; std::cout << "b_k_n: " << b_k_n.mDesc << std::endl;
......
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