Skip to content
This repository was archived by the owner on Aug 21, 2025. It is now read-only.

Commit 557a7b3

Browse files
committed
fix ravel part 2
1 parent 78ae0c6 commit 557a7b3

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

test/test_vmap.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -170,8 +170,9 @@ def test_unsupported_op_err_msg(self):
170170
r"Batching rule not implemented for aten::.+; the "
171171
r"fallback path doesn't work on out= or view ops"
172172
)
173-
with self.assertRaisesRegex(RuntimeError, msg):
174-
vmap(torch.ravel)(tensor)
173+
# TODO: find a view op
174+
# with self.assertRaisesRegex(RuntimeError, msg):
175+
# vmap(torch.ravel)(tensor)
175176

176177
def out_op(x, y):
177178
return torch.abs(x, out=y)

0 commit comments

Comments
 (0)