Commit 84e198a3 authored by wenjh's avatar wenjh
Browse files

Merge branch 'develop_v2.4'

parents 638296df 7d2b9c77
......@@ -116,11 +116,11 @@ void performTest(bool use_bias, bool use_gelu, const size_t m, const size_t k, c
Tensor D("D", std::vector<size_t>{ n, m }, dtype);
Tensor bias;
if(use_bias){
bias = Tensor("bias", {m}, bias_type);
bias = Tensor("bias", std::vector<size_t>{m}, bias_type);
}
Tensor pre_gelu_out;
if(use_gelu){
pre_gelu_out = Tensor("pre_gelu_out", { n, m }, gelu_type);
pre_gelu_out = Tensor("pre_gelu_out", std::vector<size_t>{ n, m }, gelu_type);
}
//initialize the data and scale inv of A, B
......
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