Skip to content

GitLab

"examples/vscode:/vscode.git/clone" did not exist on "c2e87869be81bef9c87adb34b7815adebe042e64"
Commit e0b0d350 authored by carlushuang's avatar carlushuang
Browse files

fix type in init

parent 9b0ee1f1
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 8 deletions
example/01_gemm/run_gemm_example.inc
View file @ e0b0d350
...@@ -65,16 +65,12 @@ bool run_gemm(const ProblemType& problem_size, const ExecutionConfig& config) ...@@ -65,16 +65,12 @@ bool run_gemm(const ProblemType& problem_size, const ExecutionConfig& config)
ck::utils::FillConstant<BDataType>{static_cast<BDataType>(1.f)}(b_k_n); ck::utils::FillConstant<BDataType>{static_cast<BDataType>(1.f)}(b_k_n);
break; break;
case 1: case 1:
ck::utils::FillUniformDistributionIntegerValue<ADataType>{ ck::utils::FillUniformDistributionIntegerValue<ADataType>{-5.f, 5.f}(a_m_k);
static_cast<ADataType>(-5.f), static_cast<ADataType>(5.f)}(a_m_k); ck::utils::FillUniformDistributionIntegerValue<BDataType>{-5.f, 5.f}(b_k_n);
ck::utils::FillUniformDistributionIntegerValue<BDataType>{
static_cast<BDataType>(-5.f), static_cast<BDataType>(5.f)}(b_k_n);
break; break;
default: default:
ck::utils::FillUniformDistribution<ADataType>{static_cast<ADataType>(-1.f), ck::utils::FillUniformDistribution<ADataType>{-1.f, 1.f}(a_m_k);
static_cast<ADataType>(1.f)}(a_m_k); ck::utils::FillUniformDistribution<BDataType>{-1.f, 1.f}(b_k_n);
ck::utils::FillUniformDistribution<BDataType>{static_cast<BDataType>(-1.f),
static_cast<BDataType>(1.f)}(b_k_n);
} }
Tensor<CDataType> 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{}));
......
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