From e4ed8569975abb514561fa2528a9ee200ac76a84 Mon Sep 17 00:00:00 2001 From: jim19930609 Date: Tue, 29 Mar 2022 08:34:17 +0000 Subject: [PATCH] Enabled final state matmul at Python API level --- python/paddle/tensor/linalg.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/python/paddle/tensor/linalg.py b/python/paddle/tensor/linalg.py index b2455323c2b2c..7901379d9c793 100644 --- a/python/paddle/tensor/linalg.py +++ b/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] """ - op_type = 'matmul_v2' - if paddle.in_dynamic_mode(): + if in_dygraph_mode(): + return _C_ops.final_state_matmul(x, y, transpose_x, transpose_y) + + if _in_legacy_dygraph(): + op_type = 'matmul_v2' op = getattr(_C_ops, op_type) return op(x, y, 'trans_x', transpose_x, 'trans_y', transpose_y)