If a torch.* returns non-Tensor, make this unimplemented rather than assert. (#89918)

Signed-off-by: Edward Z. Yang <ezyang@fb.com>

Pull Request resolved: https://github.com/pytorch/pytorch/pull/89918
Approved by: https://github.com/albanD
This commit is contained in:
Edward Z. Yang
2022-12-16 05:46:27 +08:00
committed by PyTorch MergeBot
parent a66af1feba
commit e686a442b4
9 changed files with 17 additions and 3 deletions

View File

@ -8,7 +8,7 @@ from typing import Union
import unittest
import torch.testing._internal.common_utils as common
from torch.testing._internal.common_utils import IS_WINDOWS
from torch.testing._internal.common_utils import IS_WINDOWS, skipIfTorchDynamo
from torch.testing._internal.common_cuda import TEST_CUDA
import torch
import torch.backends.cudnn
@ -295,6 +295,7 @@ class TestRNGExtension(common.TestCase):
def setUp(self):
super(TestRNGExtension, self).setUp()
@skipIfTorchDynamo("https://github.com/pytorch/torchdynamo/issues/1991")
def test_rng(self):
fourty_two = torch.full((10,), 42, dtype=torch.int64)