add test
This commit is contained in:
parent
145f09fc92
commit
8ad3edfc40
1 changed files with 21 additions and 0 deletions
|
@ -1063,6 +1063,26 @@ struct test_sqr : public test_case {
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
// GGML_OP_SQRT
|
||||||
|
struct test_sqrt : public test_case {
|
||||||
|
const ggml_type type;
|
||||||
|
const std::array<int64_t, 4> ne;
|
||||||
|
|
||||||
|
std::string vars() override {
|
||||||
|
return VARS_TO_STR2(type, ne);
|
||||||
|
}
|
||||||
|
|
||||||
|
test_sqrt(ggml_type type = GGML_TYPE_F32,
|
||||||
|
std::array<int64_t, 4> ne = {10, 10, 10, 10})
|
||||||
|
: type(type), ne(ne) {}
|
||||||
|
|
||||||
|
ggml_tensor * build_graph(ggml_context * ctx) override {
|
||||||
|
ggml_tensor * a = ggml_new_tensor(ctx, type, 4, ne.data());
|
||||||
|
ggml_tensor * out = ggml_sqrt(ctx, a);
|
||||||
|
return out;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
// GGML_OP_CLAMP
|
// GGML_OP_CLAMP
|
||||||
struct test_clamp : public test_case {
|
struct test_clamp : public test_case {
|
||||||
const ggml_type type;
|
const ggml_type type;
|
||||||
|
@ -2200,6 +2220,7 @@ static bool test_backend(ggml_backend_t backend, test_mode mode, const char * op
|
||||||
}
|
}
|
||||||
|
|
||||||
test_cases.emplace_back(new test_sqr());
|
test_cases.emplace_back(new test_sqr());
|
||||||
|
test_cases.emplace_back(new test_sqrt());
|
||||||
test_cases.emplace_back(new test_clamp());
|
test_cases.emplace_back(new test_clamp());
|
||||||
|
|
||||||
test_cases.emplace_back(new test_diag_mask_inf(GGML_TYPE_F32, {10, 10, 1, 1}, 5));
|
test_cases.emplace_back(new test_diag_mask_inf(GGML_TYPE_F32, {10, 10, 1, 1}, 5));
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue