Skip to content
This repository has been archived by the owner on Jan 24, 2024. It is now read-only.

Commit

Permalink
fix bug
Browse files Browse the repository at this point in the history
  • Loading branch information
@zrr1999
zrr1999 committed Aug 21, 2022
1 parent 9527560 commit 0c488c4
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions cinn/frontend/net_builder_test.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -239,7 +239,7 @@ TEST(net_build, program_execute_gather) {

auto input1_tensor = scope->GetTensor(std::string(input1.id()));
SetFloatRandData(input1_tensor, target);
int* input1_data = input1_tensor->mutable_data<float>(target);
float* input1_data = input1_tensor->mutable_data<float>(target);

auto input2_tensor = scope->GetTensor(std::string(input2.id()));
SetIntRandData(input2_tensor, target);
Expand Down Expand Up @@ -293,7 +293,7 @@ TEST(net_build, program_execute_gather_nd) {

auto input1_tensor = scope->GetTensor(std::string(input1.id()));
SetFloatRandData(input1_tensor, target);
int* input1_data = input1_tensor->mutable_data<float>(target);
float* input1_data = input1_tensor->mutable_data<float>(target);

auto input2_tensor = scope->GetTensor(std::string(input2.id()));
SetIntRandData(input2_tensor, target);
Expand Down

0 comments on commit 0c488c4

Please sign in to comment.