Skip to content

Commit

Permalink
delete dygraph in unittest
Browse files Browse the repository at this point in the history
  • Loading branch information
pangyoki committed Mar 30, 2021
1 parent fa651ac commit bfcc7cd
Showing 1 changed file with 0 additions and 6 deletions.
6 changes: 0 additions & 6 deletions python/paddle/fluid/tests/unittests/npu/test_save_load_npu.py
Original file line number Diff line number Diff line change
Expand Up @@ -1314,12 +1314,6 @@ def symlink_force(target, link_name):
fluid.set_program_state(main_program, program_state)
self.check_in_static(main_program, base_map)

# make sure `load_program_state` can be used in dynamic graph mode
with fluid.dygraph.guard(place):
load_state = fluid.load_program_state("test_program_1")
for k, v in load_state.items():
self.assertTrue(np.array_equal(base_map[k], v))

def check_in_static(self, main_program, base_map):
for var in main_program.list_vars():
if isinstance(var, framework.Parameter) or var.persistable:
Expand Down

0 comments on commit bfcc7cd

Please sign in to comment.