File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -270,7 +270,7 @@ def forward(self, x):
270270 )
271271
272272 # Save a copy of the edge program since to_executorch is
273- # stateful to sombe degree.
273+ # stateful to some degree.
274274 edge_copy = deepcopy (edge )
275275 et_config = ExecutorchBackendConfig (
276276 passes = [InitializedMutableBufferPass (["cache_pos" ])],
@@ -289,7 +289,8 @@ def forward(self, x):
289289 torch .allclose (
290290 method_init_pass .execute ((example_inputs ))[0 ], torch .arange (1 , 11 )
291291 )
292- # Test that the mutable buffer is uninitialized and starts with default zeros.
292+ # Test that the mutable buffer is uninitialized and starts with default zeros,
293+ # we test equality with torch.ones because of the mutation += 1 in the model forward.
293294 torch .allclose (
294295 method_regular .execute ((example_inputs ))[0 ],
295296 torch .ones (10 , dtype = torch .int64 ),
You can’t perform that action at this time.
0 commit comments