p60419572
/
InfiniTensor
forked from jiuyuan/InfiniTensor
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

update test

This commit is contained in:
Bolun 2023-10-12 14:07:23 +08:00
parent c774c9182d
commit ee6dd3deac
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/kernels/bang/test_bang_codegen.cc
View File

@ -56,7 +56,7 @@ TEST(BANG_Codegen, run) {
auto temp3 = sqrt->getOutput(); auto temp3 = sqrt->getOutput();
auto mul = g->addOp<MulObj>(d, temp3, nullptr); auto mul = g->addOp<MulObj>(d, temp3, nullptr);
auto temp4 = mul->getOutput(); auto temp4 = mul->getOutput();
auto softmax = g->addOp<SigmoidObj>(temp4, nullptr, 0); auto softmax = g->addOp<SigmoidObj>(temp4, nullptr);
// allocate BANG memory // allocate BANG memory
g->dataMalloc(); g->dataMalloc();

2
test/kernels/cuda/test_cuda_codegen.cc
View File

@ -57,7 +57,7 @@ TEST(CUDA_Codegen, run) {
auto temp3 = sqrt->getOutput(); auto temp3 = sqrt->getOutput();
auto mul = g->addOp<MulObj>(d, temp3, nullptr); auto mul = g->addOp<MulObj>(d, temp3, nullptr);
auto temp4 = mul->getOutput(); auto temp4 = mul->getOutput();
auto softmax = g->addOp<SigmoidObj>(temp4, nullptr, 0); auto softmax = g->addOp<SigmoidObj>(temp4, nullptr);
// allocate CUDA memory // allocate CUDA memory
g->dataMalloc(); g->dataMalloc();