【Hackathon 9th No.118】Convert torch._C._nn.softplus to torch.nn.functional.softplus (#334)

* Convert torch._C._nn.softplus to torch.nn.functional.softplus

* resolve merge conflicts

* resolve merge conflicts

Author: G2uge

graph_net/torch/backend/unstable_to_stable_backend.py

@@ -154,14 +154,32 @@ def _impl_unstable_to_stable_special_logit(self, gm):
 
         # Recompile the graph
         gm.recompile()
-
         return gm
 
     # replace this line with modification code for task 116 (torch._C._linalg.linalg_vector_norm)
 
     # replace this line with modification code for task 117 (torch._C._linalg.linalg_norm)
 
-    # replace this line with modification code for task 118 (torch._C._nn.softplus)
+    def _impl_unstable_to_stable_softplus(self, gm):
+        """
+        Convert torch._C._nn.softplus to torch.nn.functional.softplus
+        """
+        import torch.nn.functional as F
+
+        issue_nodes = (
+            node
+            for node in gm.graph.nodes
+            if node.op == "call_function"
+            if hasattr(node.target, "__module__")
+            if node.target.__module__ == "torch._C._nn"
+            if hasattr(node.target, "__name__")
+            if node.target.__name__ == "softplus"
+        )
+        for node in issue_nodes:
+            node.target = F.softplus
+
+        gm.recompile()
+        return gm
 
     # replace this line with modification code for task 119 (torch._C._nn.one_hot)
 

graph_net/torch/fx_graph_serialize_util.py

@@ -141,7 +141,7 @@ def serialize_graph_module_to_str(gm: torch.fx.GraphModule) -> str:
         (r"torch\._C\._special\.special_logit\(", "torch.special.logit("),
         # replace this line with modification code for task 116 (torch._C._linalg.linalg_vector_norm)
         # replace this line with modification code for task 117 (torch._C._linalg.linalg_norm)
-        # replace this line with modification code for task 118 (torch._C._nn.softplus)
+        (r"torch\._C\._nn\.softplus\(", "torch.nn.functional.softplus("),
         # replace this line with modification code for task 119 (torch._C._nn.one_hot)
         # replace this line with modification code for task 121 (torch._C._set_grad_enabled)
         # replace this line with modification code for task 122 (torch._C._log_api_usage_once)