remove max inplace grad add

pull/3346/head
phlrain 2025-01-26 15:15:20 +08:00
parent 5580de6077
commit 7bd58aa730
2 changed files with 1 additions and 2 deletions

View File

@ -645,7 +645,7 @@ class Engine(object):
self.auto_cast = AutoCast(use_amp)
self.scaler = build_scaler(use_amp)
else:
AMP_RELATED_FLAGS_SETTING = {'FLAGS_max_inplace_grad_add': 8, }
AMP_RELATED_FLAGS_SETTING = {}
if paddle.is_compiled_with_cuda():
AMP_RELATED_FLAGS_SETTING.update({
'FLAGS_cudnn_batchnorm_spatial_persistent': 1

View File

@ -104,7 +104,6 @@ def main(args):
'FLAGS_cudnn_exhaustive_search': 1,
'FLAGS_conv_workspace_size_limit': 1500,
'FLAGS_cudnn_batchnorm_spatial_persistent': 1,
'FLAGS_max_inplace_grad_add': 8,
}
os.environ['FLAGS_cudnn_batchnorm_spatial_persistent'] = '1'
paddle.set_flags(AMP_RELATED_FLAGS_SETTING)