diff --git a/python/paddle/fluid/contrib/mixed_precision/fp16_lists.py b/python/paddle/fluid/contrib/mixed_precision/fp16_lists.py index a9f080c514..8c467a4969 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 ffb4d9f16f..4ff0833787 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', }