"...resnet50_tensorflow.git" did not exist on "8060872c96e71c6356d7bf6de1e5e7bb4d88888c"
Unverified Commit 8a2c69ee authored by Chao Liu's avatar Chao Liu Committed by GitHub
Browse files

use integer value for GEMM test (#219)

parent c77ae65d
......@@ -139,17 +139,10 @@ struct TestGemm
Tensor<CDataType> c_m_n_device_result(
f_host_tensor_descriptor(params.M, params.N, params.StrideC, CLayout{}));
auto f_generate_tensor_value = [](auto& desc, auto type) {
auto f_generate_tensor_value = [](auto& tensor, auto type) {
using dataType = decltype(type);
if(std::is_same<dataType, int8_t>::value)
{
desc.GenerateTensorValue(GeneratorTensor_2<int8_t>{-5, 5});
}
else
{
desc.GenerateTensorValue(GeneratorTensor_3<dataType>{-0.5, 0.5});
}
tensor.GenerateTensorValue(GeneratorTensor_2<dataType>{-5, 5});
};
f_generate_tensor_value(a_m_k, ADataType{});
......
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