update test fixture (#89796)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/89796
Approved by: https://github.com/davidberard98
This commit is contained in:
Elias Ellison
2023-02-03 18:56:57 +00:00
committed by PyTorch MergeBot
parent 513b5da357
commit a2ac25f63e
5 changed files with 31 additions and 0 deletions

Binary file not shown.

Binary file not shown.

Binary file not shown.

View File

@ -57,3 +57,31 @@ class TestVersionedGeluOutV9(torch.nn.Module):
def forward(self, x):
out = torch.zeros_like(x)
return torch._C._nn.gelu(x, out=out)
class TestVersionedRandomV10(torch.nn.Module):
def __init__(self):
super().__init__()
def forward(self, x):
out = torch.zeros_like(x)
return out.random_(0, 10)
class TestVersionedRandomFuncV10(torch.nn.Module):
def __init__(self):
super().__init__()
def forward(self, x):
out = torch.zeros_like(x)
return out.random(0, 10)
class TestVersionedRandomOutV10(torch.nn.Module):
def __init__(self):
super().__init__()
def forward(self, x):
x = torch.zeros_like(x)
out = torch.zeros_like(x)
x.random(0, 10, out=out)
return out

View File

@ -96,6 +96,9 @@ ALL_MODULES = {
TestVersionedLogspaceOutV8(): "aten::logspace.out",
TestVersionedGeluV9(): "aten::gelu",
TestVersionedGeluOutV9(): "aten::gelu.out",
TestVersionedRandomV10(): "aten::random_.from",
TestVersionedRandomFuncV10(): "aten::random.from",
TestVersionedRandomOutV10(): "aten::random.from_out",
}
"""