mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
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:
committed by
PyTorch MergeBot
parent
a66af1feba
commit
e686a442b4
@ -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)
|
||||
|
||||
|
Reference in New Issue
Block a user