[Prim] fix p_norm decomp (#71218)

zhanghonggeng · web-flow · commit 7b3931027991 · 2025-02-21T16:47:13.000+08:00
diff --git a/paddle/fluid/primitive/decomp_rule/decomp_rule/composite.h b/paddle/fluid/primitive/decomp_rule/decomp_rule/composite.h
@@ -148,7 +148,7 @@ Tensor p_norm_decomp(const Tensor& x,
     auto porder_tensor = full_scalar<T>(porder, x_tmp.dtype(), x_tmp.place());
     auto inv_porder_tensor =
         full_scalar<T>(1 / porder, x_tmp.dtype(), x_tmp.place());
-    res = elementwise_pow<T>(x_tmp, porder_tensor);
+    res = elementwise_pow<T>(abs<T>(x_tmp), porder_tensor);
     res = sum<T>(res, reduce_axis, x_tmp.dtype(), keepdim);
     res = elementwise_pow<T>(res, inv_porder_tensor);
   }

Original file line number	Diff line number	Diff line change
`@@ -148,7 +148,7 @@ Tensor p_norm_decomp(const Tensor& x,`
`148`	`148`	`auto porder_tensor = full_scalar<T>(porder, x_tmp.dtype(), x_tmp.place());`
`149`	`149`	`auto inv_porder_tensor =`
`150`	`150`	`full_scalar<T>(1 / porder, x_tmp.dtype(), x_tmp.place());`
`151`		`- res = elementwise_pow<T>(x_tmp, porder_tensor);`
	`151`	`+ res = elementwise_pow<T>(abs<T>(x_tmp), porder_tensor);`
`152`	`152`	`res = sum<T>(res, reduce_axis, x_tmp.dtype(), keepdim);`
`153`	`153`	`res = elementwise_pow<T>(res, inv_porder_tensor);`
`154`	`154`	`}`