Skip to content

Commit

Permalink
Revert some changes.
Browse files Browse the repository at this point in the history
  • Loading branch information
arogowie-intel committed Aug 23, 2021
1 parent 89e4d3d commit a843a50
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions python/paddle/fluid/contrib/mixed_precision/bf16/amp_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -128,8 +128,8 @@ def _insert_cast_op(block, op, idx, src_dtype, dest_dtype):
continue
if out_var.dtype == core.VarDesc.VarType.FP32:
out_var.desc.set_dtype(core.VarDesc.VarType.BF16)
if op.has_attr('out_dtype'):
op._set_attr('out_dtype', core.VarDesc.VarType.BF16)
if op.has_attr('out_dtype'):
op._set_attr('out_dtype', core.VarDesc.VarType.BF16)
return num_cast_ops


Expand Down
2 changes: 1 addition & 1 deletion python/paddle/fluid/tests/book/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -12,4 +12,4 @@ set_tests_properties(test_image_classification PROPERTIES TIMEOUT 200)
set_tests_properties(test_label_semantic_roles PROPERTIES TIMEOUT 240)
set_tests_properties(test_machine_translation PROPERTIES TIMEOUT 120)
set_tests_properties(test_rnn_encoder_decoder PROPERTIES TIMEOUT 120)
set_tests_properties(test_fit_a_line PROPERTIES TIMEOUT 140)
set_tests_properties(test_fit_a_line PROPERTIES TIMEOUT 120)

0 comments on commit a843a50

Please sign in to comment.