Skip to content

Commit 6bb6ab5

Browse files
authored
Update dynamo cpu fallback op to aten::_foobar (#5393)
1 parent f5b6f1c commit 6bb6ab5

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

test/dynamo/test_dynamo.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)