diff --git a/test/jit/fixtures/test_versioned_gelu_out_v9.ptl b/test/jit/fixtures/test_versioned_gelu_out_v9.ptl new file mode 100644 index 00000000000..208ae510075 Binary files /dev/null and b/test/jit/fixtures/test_versioned_gelu_out_v9.ptl differ diff --git a/test/jit/fixtures/test_versioned_gelu_v9.ptl b/test/jit/fixtures/test_versioned_gelu_v9.ptl new file mode 100644 index 00000000000..5e4ffb20f82 Binary files /dev/null and b/test/jit/fixtures/test_versioned_gelu_v9.ptl differ diff --git a/test/jit/fixtures_srcs/fixtures_src.py b/test/jit/fixtures_srcs/fixtures_src.py index 545152b6a3a..dff23702311 100644 --- a/test/jit/fixtures_srcs/fixtures_src.py +++ b/test/jit/fixtures_srcs/fixtures_src.py @@ -42,3 +42,18 @@ def __init__(self): def forward(self, a: Union[int, float, complex], b: Union[int, float, complex], out: torch.Tensor): return torch.logspace(a, b, out=out) + +class TestVersionedGeluV9(torch.nn.Module): + def __init__(self): + super().__init__() + + def forward(self, x): + return torch._C._nn.gelu(x) + +class TestVersionedGeluOutV9(torch.nn.Module): + def __init__(self): + super().__init__() + + def forward(self, x): + out = torch.zeros_like(x) + return torch._C._nn.gelu(x, out=out) diff --git a/test/jit/fixtures_srcs/generate_models.py b/test/jit/fixtures_srcs/generate_models.py index 36b6b5ffe68..980e7dd0324 100644 --- a/test/jit/fixtures_srcs/generate_models.py +++ b/test/jit/fixtures_srcs/generate_models.py @@ -94,6 +94,8 @@ def div_Tensor_0_3(self: Tensor, other: Tensor) -> Tensor: TestVersionedLinspaceOutV7(): "aten::linspace.out", TestVersionedLogspaceV8(): "aten::logspace", TestVersionedLogspaceOutV8(): "aten::logspace.out", + TestVersionedGeluV9(): "aten::gelu", + TestVersionedGeluOutV9(): "aten::gelu.out", } """