"csrc/vscode:/vscode.git/clone" did not exist on "7e1d5e5308fa3549dfed1821188d588260a03c8a"
Commit 7d2b9c77 authored by wenjh's avatar wenjh
Browse files

Fix build error of test_cublaslt_gemm


Signed-off-by: wenjh's avatarwenjh <wenjh@sugon.com>
parent 6d461a10
...@@ -116,11 +116,11 @@ void performTest(bool use_bias, bool use_gelu, const size_t m, const size_t k, c ...@@ -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 D("D", std::vector<size_t>{ n, m }, dtype);
Tensor bias; Tensor bias;
if(use_bias){ if(use_bias){
bias = Tensor("bias", {m}, bias_type); bias = Tensor("bias", std::vector<size_t>{m}, bias_type);
} }
Tensor pre_gelu_out; Tensor pre_gelu_out;
if(use_gelu){ 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 //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