diff --git a/python/paddle/fluid/contrib/mixed_precision/fp16_lists.py b/python/paddle/fluid/contrib/mixed_precision/fp16_lists.py index a9f080c514dff078b0068bce262fa177fd0b0db2..8c467a4969e295340d626b96294611ce50207d74 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 ffb4d9f16f29f384b83f175ddcb60f65e8077930..4ff08337875c030827a762eb4199c1a1e28781e4 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', }