File tree Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Original file line number Diff line number Diff line change @@ -441,6 +441,10 @@ void* ggml_cuda_cpy_fn(const ggml_tensor * src0, ggml_tensor * src1) {
441441 return (void *) cpy_flt<cpy_1_flt<nv_bfloat16, nv_bfloat16>>;
442442 } else if (src0->type == GGML_TYPE_BF16 && src1->type == GGML_TYPE_F32) {
443443 return (void *) cpy_flt<cpy_1_flt<nv_bfloat16, float >>;
444+ } else if (src0->type == GGML_TYPE_F32 && src1->type == GGML_TYPE_I32) {
445+ return (void *) cpy_flt<cpy_1_flt<float , int32_t >>;
446+ } else if (src0->type == GGML_TYPE_I32 && src1->type == GGML_TYPE_F32) {
447+ return (void *) cpy_flt<cpy_1_flt<int32_t , float >>;
444448 } else {
445449 GGML_ABORT (" %s: unsupported type combination (%s to %s)\n " , __func__,
446450 ggml_type_name (src0->type ), ggml_type_name (src1->type ));
You can’t perform that action at this time.
0 commit comments