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

fix

This commit is contained in:
Bolun 2023-10-12 09:59:08 +08:00
parent 764702beb2
commit c774c9182d
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -40,7 +40,7 @@ TEST(BANG_Codegen, run) {
aCpu->setData(IncrementalGenerator());
bCpu->setData(IncrementalGenerator());
cCpu->setData(OneGenerator());
dCpu->setData(OneGenerator());
dCpu->setData(IncrementalGenerator());
// Build BANG graph
Graph g = make_ref<GraphObj>(bangRuntime);
@ -56,7 +56,7 @@ TEST(BANG_Codegen, run) {
auto temp3 = sqrt->getOutput();
auto mul = g->addOp<MulObj>(d, temp3, nullptr);
auto temp4 = mul->getOutput();
auto softmax = g->addOp<SoftmaxObj>(temp4, nullptr, 0);
auto softmax = g->addOp<SigmoidObj>(temp4, nullptr, 0);
// allocate BANG memory
g->dataMalloc();

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

@ -41,7 +41,7 @@ TEST(CUDA_Codegen, run) {
aCpu->setData(IncrementalGenerator());
bCpu->setData(IncrementalGenerator());
cCpu->setData(OneGenerator());
dCpu->setData(OneGenerator());
dCpu->setData(IncrementalGenerator());
// Build CUDA graph
Graph g = make_ref<GraphObj>(cudaRuntime);
@ -57,7 +57,7 @@ TEST(CUDA_Codegen, run) {
auto temp3 = sqrt->getOutput();
auto mul = g->addOp<MulObj>(d, temp3, nullptr);
auto temp4 = mul->getOutput();
auto softmax = g->addOp<SoftmaxObj>(temp4, nullptr, 0);
auto softmax = g->addOp<SigmoidObj>(temp4, nullptr, 0);
// allocate CUDA memory
g->dataMalloc();