Enabled final state matmul at Python API level (#41089)

python/paddle/tensor/linalg.py

@@ -133,8 +133,11 @@ def matmul(x, y, transpose_x=False, transpose_y=False, name=None):
         # [10, 3, 5, 5]
 
     """
+    if in_dygraph_mode():
+        return _C_ops.final_state_matmul(x, y, transpose_x, transpose_y)
+
+    if _in_legacy_dygraph():
         op_type = 'matmul_v2'
-    if paddle.in_dynamic_mode():
         op = getattr(_C_ops, op_type)
         return op(x, y, 'trans_x', transpose_x, 'trans_y', transpose_y)