MCPcopy Create free account
hub / github.com/Tiiny-AI/PowerInfer / ggml_can_mul_mat

Function ggml_can_mul_mat

ggml.c:2035–2041  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

2033}
2034
2035static inline bool ggml_can_mul_mat(const struct ggml_tensor * t0, const struct ggml_tensor * t1) {
2036 static_assert(GGML_MAX_DIMS == 4, "GGML_MAX_DIMS is not 4 - update this function");
2037
2038 return (t0->ne[0] == t1->ne[0]) &&
2039 (t1->ne[2]%t0->ne[2] == 0) && // verify t0 is broadcastable
2040 (t1->ne[3]%t0->ne[3] == 0);
2041}
2042
2043static inline bool ggml_can_out_prod(const struct ggml_tensor * t0, const struct ggml_tensor * t1) {
2044 static_assert(GGML_MAX_DIMS == 4, "GGML_MAX_DIMS is not 4 - update this function");

Callers 4

ggml_mul_matFunction · 0.70
ggml_flash_attnFunction · 0.70
ggml_flash_ffFunction · 0.70
ggml_flash_attn_backFunction · 0.70

Calls

no outgoing calls

Tested by

no test coverage detected