Skip to content

Commit

Permalink
fix cpp tests
Browse files Browse the repository at this point in the history
fix
  • Loading branch information
mehrdadh committed Aug 12, 2022
1 parent 6258b0b commit 4f21a7b
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 7 deletions.
10 changes: 5 additions & 5 deletions tests/cpp/c_codegen_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ TEST(CCodegen, MainFunctionOrder) {

std::string tvm_module_main = std::string(runtime::symbol::tvm_module_main);

tvm::Target target_c = tvm::Target("c -keys=cpu -link-params=0");
tvm::Target target_c = tvm::Target("c -keys=cpu");

const int n = 4;
Array<PrimExpr> shape{n};
Expand Down Expand Up @@ -104,16 +104,16 @@ TEST(CCodegen, FunctionOrder) {
using namespace tvm;
using namespace tvm::te;

Target target = Target("c -keys=cpu -link-params=0");
Target target = Target("c -keys=cpu");

// add schedules in reverse order
Map<tvm::Target, IRModule> inputs;
inputs.Set(Target("c -keys=cpu -link-params=0"), BuildLowered("op_2", target));
inputs.Set(Target("c -keys=cpu -link-params=0"), BuildLowered("op_1", target));
inputs.Set(Target("c -keys=cpu"), BuildLowered("op_2", target));
inputs.Set(Target("c -keys=cpu"), BuildLowered("op_1", target));

for (uint32_t counter = 99; IsSorted(inputs) && counter > 0; counter--) {
std::string op_name = "op_" + std::to_string(counter);
inputs.Set(Target("c -keys=cpu -link-params=0"), BuildLowered(op_name, target));
inputs.Set(Target("c -keys=cpu"), BuildLowered(op_name, target));
}

EXPECT_FALSE(IsSorted(inputs));
Expand Down
2 changes: 0 additions & 2 deletions tests/cpp/target_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -336,7 +336,6 @@ TEST(TargetCreation, DeduplicateKeys) {
ICHECK_EQ(target->keys[1], "arm_cpu");
ICHECK_EQ(target->attrs.size(), 2U);
ICHECK_EQ(target->GetAttr<String>("device"), "arm_cpu");
ICHECK_EQ(target->GetAttr<Bool>("link-params"), false);
}

TEST(TargetKindRegistry, ListTargetKinds) {
Expand All @@ -352,5 +351,4 @@ TEST(TargetKindRegistry, ListTargetOptions) {

ICHECK_EQ(attrs["mattr"], "Array");
ICHECK_EQ(attrs["mcpu"], "runtime.String");
ICHECK_EQ(attrs["system-lib"], "IntImm");
}

0 comments on commit 4f21a7b

Please sign in to comment.