diff --git a/graph_net/torch/backend/unstable_to_stable_backend.py b/graph_net/torch/backend/unstable_to_stable_backend.py index 7bbf588fd..f8cb3969c 100644 --- a/graph_net/torch/backend/unstable_to_stable_backend.py +++ b/graph_net/torch/backend/unstable_to_stable_backend.py @@ -147,6 +147,26 @@ def _impl_unstable_to_stable_special_logit(self, gm): 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) + + # 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) + + # replace this line with modification code for task 123 (torch._C._nn.pad) + + # replace this line with modification code for task 125 (torch._C._nn.gelu) + + # replace this line with modification code for task 126 (torch._C._nn.scaled_dot_product_attention) + + # replace this line with modification code for task 127 (torch._C._nn.linear) + def unstable_to_stable(self, gm): methods = ( name diff --git a/graph_net/torch/fx_graph_serialize_util.py b/graph_net/torch/fx_graph_serialize_util.py index 0814eef93..a67f83882 100644 --- a/graph_net/torch/fx_graph_serialize_util.py +++ b/graph_net/torch/fx_graph_serialize_util.py @@ -25,7 +25,16 @@ def serialize_graph_module_to_str(gm: torch.fx.GraphModule) -> str: (r"torch\._C\._fft\.fft_rfft\(", "torch.fft.rfft("), (r"torch\._C\._fft\.fft_fftn\(", "torch.fft.fftn("), (r"torch\._C\._special\.special_logit\(", "torch.special.logit("), - # Add new rules to this list as needed + # 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) + # 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) + # replace this line with modification code for task 123 (torch._C._nn.pad) + # replace this line with modification code for task 125 (torch._C._nn.gelu) + # replace this line with modification code for task 126 (torch._C._nn.scaled_dot_product_attention) + # replace this line with modification code for task 127 (torch._C._nn.linear) ] for pattern, repl in replacements: code = re.sub(pattern, repl, code)