happyz
/
llama.cpp
mirror of https://github.com/ggerganov/llama.cpp.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

one more CI fix

This commit is contained in:
bssrdf 2025-11-03 09:30:51 -05:00
parent 5aa4ae739d
commit 91650b7fdc
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test-conv3d.cpp
View File

@ -37,6 +37,9 @@ struct test_model {
struct ggml_context * ctx; struct ggml_context * ctx;
}; };
void load_model(test_model & model, int ic, int oc, int iw, int ih, int id,
int kw = 3, int kh = 3, int kd = 3,
bool use_fp16 = true, bool use_gpu = false );
struct ggml_cgraph * build_graph_0(const test_model& model, const int64_t ic, const int64_t n, const int64_t oc); struct ggml_cgraph * build_graph_0(const test_model& model, const int64_t ic, const int64_t n, const int64_t oc);
struct ggml_cgraph * build_graph_1(const test_model& model, const int64_t ic, const int64_t n, const int64_t oc); struct ggml_cgraph * build_graph_1(const test_model& model, const int64_t ic, const int64_t n, const int64_t oc);
typedef struct ggml_cgraph* (*build_graph_t)(const test_model& model, typedef struct ggml_cgraph* (*build_graph_t)(const test_model& model,
@ -49,7 +52,7 @@ std::vector<float> compute_graph(const test_model & model, ggml_gallocr_t allocr
void load_model(test_model & model, int ic, int oc, int iw, int ih, int id, void load_model(test_model & model, int ic, int oc, int iw, int ih, int id,
int kw = 3, int kh = 3, int kd = 3, int kw = 3, int kh = 3, int kd = 3,
bool use_fp16 = true, bool use_gpu = false ) { bool use_fp16 = true, bool use_gpu = false ) {
// create data // create data
int KW = kw, KH = kh, KD = kd; int KW = kw, KH = kh, KD = kd;
int IC = ic, OC = oc; int IC = ic, OC = oc;