From 2755ae6a2ab6ac70e40397439de62faefda2ffcc Mon Sep 17 00:00:00 2001 From: Aurelius84 Date: Tue, 11 Oct 2022 19:20:03 +0800 Subject: [PATCH] [Dy2St]Remove usless cast operation to speed up FP16 training (#46851) --- .../fluid/dygraph/dygraph_to_static/partial_program.py | 10 ---------- 1 file changed, 10 deletions(-) diff --git a/python/paddle/fluid/dygraph/dygraph_to_static/partial_program.py b/python/paddle/fluid/dygraph/dygraph_to_static/partial_program.py index 77d6104982..8592e9e891 100644 --- a/python/paddle/fluid/dygraph/dygraph_to_static/partial_program.py +++ b/python/paddle/fluid/dygraph/dygraph_to_static/partial_program.py @@ -598,16 +598,6 @@ class PartialProgramLayer: == paddle.float16): in_vars[i] = var.astype('float16') in_vars[i].name = name - if (self.forward_program.global_block().has_var(name) - and self.forward_program.global_block().var(name).dtype - == paddle.float16): - in_vars[i] = var.astype('float16') - in_vars[i].name = name - if (self.backward_program.global_block().has_var(name) - and self.backward_program.global_block().var(name).dtype - == paddle.float16): - in_vars[i] = var.astype('float16') - in_vars[i].name = name @property def program(self): -- GitLab