From 11e32baf1e0f83374e3563e7541aab085e18b1cc Mon Sep 17 00:00:00 2001
From: Leo Chen <chenqiuliang@baidu.com>
Date: Wed, 18 Nov 2020 00:18:17 +0800
Subject: [PATCH] Add matmtl_v2 to amp list (#28693)

* add matmtl_v2 to amp list

* support dygraph
---
 python/paddle/fluid/contrib/mixed_precision/fp16_lists.py | 1 +
 python/paddle/fluid/dygraph/amp/auto_cast.py              | 1 +
 2 files changed, 2 insertions(+)

diff --git a/python/paddle/fluid/contrib/mixed_precision/fp16_lists.py b/python/paddle/fluid/contrib/mixed_precision/fp16_lists.py
index a9f080c514d..8c467a4969e 100644
--- a/python/paddle/fluid/contrib/mixed_precision/fp16_lists.py
+++ b/python/paddle/fluid/contrib/mixed_precision/fp16_lists.py
@@ -74,6 +74,7 @@ class AutoMixedPrecisionLists(object):
 white_list = {
     'conv2d',
     'matmul',
+    'matmul_v2',
     'mul',
 }
 
diff --git a/python/paddle/fluid/dygraph/amp/auto_cast.py b/python/paddle/fluid/dygraph/amp/auto_cast.py
index ffb4d9f16f2..4ff08337875 100644
--- a/python/paddle/fluid/dygraph/amp/auto_cast.py
+++ b/python/paddle/fluid/dygraph/amp/auto_cast.py
@@ -27,6 +27,7 @@ __all__ = ['amp_guard']
 WHITE_LIST = {
     'conv2d',
     'matmul',
+    'matmul_v2',
     'mul',
 }
 
-- 
GitLab