未验证 提交 11e32baf 编写于 作者: L Leo Chen 提交者: GitHub

Add matmtl_v2 to amp list (#28693)

* add matmtl_v2 to amp list

* support dygraph
上级 b6f86b84
...@@ -74,6 +74,7 @@ class AutoMixedPrecisionLists(object): ...@@ -74,6 +74,7 @@ class AutoMixedPrecisionLists(object):
white_list = { white_list = {
'conv2d', 'conv2d',
'matmul', 'matmul',
'matmul_v2',
'mul', 'mul',
} }
......
...@@ -27,6 +27,7 @@ __all__ = ['amp_guard'] ...@@ -27,6 +27,7 @@ __all__ = ['amp_guard']
WHITE_LIST = { WHITE_LIST = {
'conv2d', 'conv2d',
'matmul', 'matmul',
'matmul_v2',
'mul', 'mul',
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册