File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -175,8 +175,8 @@ class DynamoCpuFallbackTest(unittest.TestCase):
175175 def test_operator_fallback (self ):
176176
177177 def fn_fallback (t ):
178- # As of 05/18/2023, torch.median is not lowered by PyTorch/XLA
179- return torch .median (t )
178+ # aten::_foobar is aux function that's used for testing purposes only
179+ return torch ._foobar (t )
180180
181181 torch ._dynamo .reset ()
182182 met .clear_counters ()
@@ -211,8 +211,8 @@ def test_fallback_multiple_submodules(self):
211211
212212 def fn_fallback (t ):
213213 t_2 = torch .mul (t , 2 )
214- # As of 05/18/2023, torch.median is not lowered by PyTorch/XLA
215- t_3 = torch .median (t_2 )
214+ # aten::_foobar is aux function that's used for testing purposes only
215+ t_3 = torch ._foobar (t_2 )
216216 t_4 = torch .mul (t_3 , 2 )
217217 return t_4
218218
You can’t perform that action at this time.
0 commit comments