Add matmtl_v2 to amp list (#28693)

* add matmtl_v2 to amp list * support dygraph

Add matmtl_v2 to amp list (#28693)
* add matmtl_v2 to amp list * support dygraph
11e32baf · Leo Chen · GitHub · b6f86b84 · 11e32baf · 11e32baf
Showing with 2 addition and 0 deletion

python/paddle/fluid/contrib/mixed_precision/fp16_lists.py python/paddle/fluid/contrib/mixed_precision/fp16_lists.py +1 -0

python/paddle/fluid/dygraph/amp/auto_cast.py python/paddle/fluid/dygraph/amp/auto_cast.py +1 -0

未找到文件。
--- 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',
 }


--- 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',
 }