Skip to content

Commit b2b9656

Browse files
committed
removed two ouputs so need to update index"
1 parent a843792 commit b2b9656

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/test_fake_tensor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1037,7 +1037,7 @@ def forward(self, arg1, arg2, arg3):
10371037
# We expect the cross ref to succed for the first output to fail
10381038
# for the rng state, see Note [Seed and Offset]
10391039
self.assertTrue("output[0]" not in str(e))
1040-
self.assertTrue("found mismatched tensor metadata for output[6]: Devices cpu and cuda:0 are not equal!" in str(e))
1040+
self.assertTrue("found mismatched tensor metadata for output[4]: Devices cpu and cuda:0 are not equal!" in str(e))
10411041

10421042
@skipIfRocm
10431043
@unittest.skipIf(not RUN_CUDA, "requires cuda")

0 commit comments

Comments
 (0)