Skip to content

Commit fd62bcc

Browse files
authored
Update unstable_to_stable_backend.py
1 parent 3a3b149 commit fd62bcc

File tree

1 file changed

+22
-21
lines changed

1 file changed

+22
-21
lines changed

graph_net/torch/backend/unstable_to_stable_backend.py

Lines changed: 22 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -157,6 +157,27 @@ def _impl_unstable_to_stable_special_logit(self, gm):
157157

158158
return gm
159159

160+
161+
162+
# replace this line with modification code for task 116 (torch._C._linalg.linalg_vector_norm)
163+
164+
# replace this line with modification code for task 117 (torch._C._linalg.linalg_norm)
165+
166+
# replace this line with modification code for task 118 (torch._C._nn.softplus)
167+
168+
# replace this line with modification code for task 119 (torch._C._nn.one_hot)
169+
170+
# replace this line with modification code for task 121 (torch._C._set_grad_enabled)
171+
172+
# replace this line with modification code for task 122 (torch._C._log_api_usage_once)
173+
174+
# replace this line with modification code for task 123 (torch._C._nn.pad)
175+
176+
# replace this line with modification code for task 125 (torch._C._nn.gelu)
177+
178+
# replace this line with modification code for task 126 (torch._C._nn.scaled_dot_product_attention)
179+
180+
160181
def _impl_unstable_to_stable_linear_to_functional_linear(self, gm):
161182
"""
162183
Convert torch._C._nn.linear to torch.nn.functional.linear
@@ -210,27 +231,7 @@ def _impl_unstable_to_stable_linear_to_functional_linear(self, gm):
210231
gm.recompile()
211232

212233
return gm
213-
214-
# replace this line with modification code for task 116 (torch._C._linalg.linalg_vector_norm)
215-
216-
# replace this line with modification code for task 117 (torch._C._linalg.linalg_norm)
217-
218-
# replace this line with modification code for task 118 (torch._C._nn.softplus)
219-
220-
# replace this line with modification code for task 119 (torch._C._nn.one_hot)
221-
222-
# replace this line with modification code for task 121 (torch._C._set_grad_enabled)
223-
224-
# replace this line with modification code for task 122 (torch._C._log_api_usage_once)
225-
226-
# replace this line with modification code for task 123 (torch._C._nn.pad)
227-
228-
# replace this line with modification code for task 125 (torch._C._nn.gelu)
229-
230-
# replace this line with modification code for task 126 (torch._C._nn.scaled_dot_product_attention)
231-
232-
# replace this line with modification code for task 127 (torch._C._nn.linear)
233-
234+
234235
def unstable_to_stable(self, gm):
235236
methods = (
236237
name

0 commit comments

Comments
 (0)