|
@@ -2351,6 +2351,7 @@ static bool ggml_cuda_compute_forward(ggml_backend_cuda_context & ctx, struct gg
|
|
|
break;
|
|
break;
|
|
|
case GGML_OP_RWKV_WKV:
|
|
case GGML_OP_RWKV_WKV:
|
|
|
ggml_cuda_op_rwkv_wkv(ctx, dst);
|
|
ggml_cuda_op_rwkv_wkv(ctx, dst);
|
|
|
|
|
+ break;
|
|
|
case GGML_OP_CROSS_ENTROPY_LOSS_BACK:
|
|
case GGML_OP_CROSS_ENTROPY_LOSS_BACK:
|
|
|
ggml_cuda_cross_entropy_loss_back(ctx, dst);
|
|
ggml_cuda_cross_entropy_loss_back(ctx, dst);
|
|
|
break;
|
|
break;
|