(uint32_t)src0->ne[0], (uint32_t)n_dims, freq_scale, (uint32_t)src0->ne[1],
freq_base, ext_factor, attn_factor, {corr_dims[0], corr_dims[1]}, theta_scale,
src2 != nullptr, (uint32_t)src0->ne[2], s1, s2,
- sections[0], sections[1], sections[2], sections[3], backprop
+ { sections[0], sections[1], sections[2], sections[3] }, backprop
}, dryrun);
}
default:
std::cerr << "ggml_vulkan: Error: Missing op: " << ggml_op_name(node->op) << std::endl;
GGML_ABORT("fatal error");
- return false;
}
vk_context compute_ctx;
default:
return false;
}
- break;
case GGML_OP_GLU:
switch (ggml_get_glu_op(op)) {
case GGML_GLU_OP_GEGLU:
default:
return false;
}
- break;
case GGML_OP_MUL_MAT:
case GGML_OP_MUL_MAT_ID:
{
}
return true;
- } break;
+ }
case GGML_OP_FLASH_ATTN_EXT:
{
ggml_backend_vk_device_context * ctx = (ggml_backend_vk_device_context *)dev->context;
default:
return false;
}
- } break;
+ }
case GGML_OP_SET_ROWS:
{
switch (op->type) {
default:
return false;
}
- } break;
+ }
case GGML_OP_CONT:
case GGML_OP_CPY:
case GGML_OP_DUP:
return true;
}
return false;
- } break;
+ }
case GGML_OP_REPEAT:
return ggml_type_size(op->type) == sizeof(float) && ggml_type_size(op->src[0]->type) == sizeof(float);
case GGML_OP_REPEAT_BACK: