fix

risemeup1 · risemeup1 · commit 6f04692f8dde · 2025-07-31T10:40:52.000+08:00
diff --git a/paddlenlp/transformers/fp8_utils.py b/paddlenlp/transformers/fp8_utils.py
@@ -89,7 +89,7 @@ def padding_and_quant_input(tensor):
             return tensor_fp8, tensor_scale, tensor_t_fp8, tensor_t_scale
 
     @staticmethod
-    def kitchen_fp8_gemm(
+    def kitchen_gemm(
         x_fp8, x_scale, w_fp8, w_scale, is_a_1d_scaled, is_b_1d_scaled, out=None, rtn_dtype=paddle.bfloat16
     ):
         if USE_DS_GEMM:
