CANN: Add L2_NORM op support (#16856)

* update L2_NORM op support * update L2_NORM op support * remove extra whitespace
2025-11-12 15:11:42 +08:00
parent 5da7664960
commit 655cddd174
3 changed files with 57 additions and 0 deletions
@@ -1777,6 +1777,9 @@ static bool ggml_cann_compute_forward(ggml_backend_cann_context & ctx, struct gg
        case GGML_OP_GROUP_NORM:
            ggml_cann_group_norm(ctx, dst);
            break;
+        case GGML_OP_L2_NORM:
+            ggml_cann_l2_norm(ctx, dst);
+            break;
        case GGML_OP_CONCAT:
            ggml_cann_concat(ctx, dst);
            break;
@@ -2515,6 +2518,7 @@ static bool ggml_backend_cann_supports_op(ggml_backend_dev_t dev, const ggml_ten
                // value of paddingW should be at most half of kernelW
                return (p0 <= (k0 / 2)) && (p1 <= (k1 / 2));
            }
+        case GGML_OP_L2_NORM:
        case GGML_OP_DUP:
        case GGML_OP_SUM:
        case GGML_OP_IM2COL: