Skip to content

【Hackathon No.89】 Remove circle import Part4 #51770

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 3 commits into from
Mar 20, 2023
Merged

【Hackathon No.89】 Remove circle import Part4 #51770

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
76833fa
use '=' replace import submodule in partial_program.py
Liyulingyue Mar 16, 2023
cd22655
roll back and test
Liyulingyue Mar 17, 2023
d5fefe9
remove all 'from paddle.amp.auto_cast import' from partial_program.py
Liyulingyue Mar 18, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 1 addition & 12 deletions python/paddle/jit/dy2static/partial_program.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@

import paddle
from paddle import _legacy_C_ops
from paddle.amp.auto_cast import _in_amp_guard, _in_pure_fp16_guard
from paddle.fluid import backward, core, framework, program_guard
from paddle.fluid.compiler import BuildStrategy
from paddle.fluid.dygraph import layers
Expand Down Expand Up @@ -455,8 +456,6 @@ def program_id(self):
"""
Return current train or eval program hash id.
"""
from paddle.amp.auto_cast import _in_amp_guard, _in_pure_fp16_guard

if self.training:
if _in_amp_guard():
return self._train_amp_program_id
Expand All @@ -474,8 +473,6 @@ def program_id(self):

@property
def train_program(self):
from paddle.amp.auto_cast import _in_amp_guard, _in_pure_fp16_guard

if _in_amp_guard():
return self._train_amp_program
elif _in_pure_fp16_guard():
Expand All @@ -485,8 +482,6 @@ def train_program(self):

@property
def infer_program(self):
from paddle.amp.auto_cast import _in_amp_guard, _in_pure_fp16_guard

if _in_amp_guard():
return self._infer_amp_program
elif _in_pure_fp16_guard():
Expand All @@ -496,8 +491,6 @@ def infer_program(self):

@property
def forward_program(self):
from paddle.amp.auto_cast import _in_amp_guard, _in_pure_fp16_guard

if self.training:
if _in_amp_guard():
progs = self._train_amp_forward_backward_program
Expand All @@ -511,8 +504,6 @@ def forward_program(self):

@property
def backward_program(self):
from paddle.amp.auto_cast import _in_amp_guard, _in_pure_fp16_guard

if self.training:
if _in_amp_guard():
progs = self._train_amp_forward_backward_program
Expand Down Expand Up @@ -708,8 +699,6 @@ def _get_double_grads(self, program):
return self._valid_vars(double_grads)

def _cast_fp16_if_pure_fp16(self, in_vars):
from paddle.amp.auto_cast import _in_pure_fp16_guard

if _in_pure_fp16_guard():
for i, var in enumerate(in_vars):
name = var.name
Expand Down