From 1681a2dde14e230a3d88288cbc4da8424b859c9c Mon Sep 17 00:00:00 2001
From: WangXi <wangxi16@baidu.com>
Date: Mon, 21 Jun 2021 16:13:42 +0800
Subject: [PATCH] update fp16 gray_list for tensor parallel (#33660)

---
 python/paddle/fluid/contrib/mixed_precision/fp16_lists.py | 2 ++
 1 file 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 2913d99ee6b..5cfa77b3d9a 100644
--- a/python/paddle/fluid/contrib/mixed_precision/fp16_lists.py
+++ b/python/paddle/fluid/contrib/mixed_precision/fp16_lists.py
@@ -146,6 +146,8 @@ gray_list = {
     'cast',
     'fused_bn_add_activation',
     'c_identity',
+    'c_concat',
+    'c_allreduce_sum',
 }
 
 # The set of ops that don't support fp16 calculation
-- 
GitLab