From 9fcb6a1d0e6872de8f7f684bc90d429e319190b6 Mon Sep 17 00:00:00 2001 From: Zhanlue Yang Date: Wed, 30 Mar 2022 10:46:33 +0800 Subject: [PATCH] Enabled final state matmul at Python API level (#41089) --- 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 b2455323c2..7901379d9c 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) -- GitLab