Handle missing symlink method on Windows (#31006)

PaddlePaddle · Feb 18, 2021 · 2497f43 · 2497f43
1 parent 5653c3a
commit 2497f43
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/python/paddle/fluid/tests/unittests/test_static_save_load.py b/python/paddle/fluid/tests/unittests/test_static_save_load.py
@@ -1276,11 +1276,11 @@ def test_ptb_rnn_cpu_float32(self):
             # case 2: load with no need file
             def symlink_force(target, link_name):
                 try:
-                    os.symlink(target, link_name)
+                    self.create_symlink(target, link_name)
                 except OSError as e:
                     if e.errno == errno.EEXIST:
                         os.remove(link_name)
-                        os.symlink(target, link_name)
+                        self.create_symlink(target, link_name)
                     else:
                         raise e
 
@@ -1304,6 +1304,14 @@ def symlink_force(target, link_name):
             for k, v in load_state.items():
                 self.assertTrue(np.array_equal(base_map[k], v))
 
+    def create_symlink(self, target, link_name):
+        try:
+            os.symlink(target, link_name)
+        except AttributeError:
+            import ctypes
+            kernel_dll = ctypes.windll.LoadLibrary("kernel32.dll")
+            kernel_dll.CreateSymbolicLinkA(target, link_name, 0)
+
     def check_in_static(self, main_program, base_map):
         for var in main_program.list_vars():
             if isinstance(var, framework.Parameter) or var.persistable: