update AOT test

Author: zezhang

test/AotCompile/BUILD

@@ -21,6 +21,7 @@ AOT_TEST_SUITE = [
     ("tanh", False),
     ("clamp", False),
     ("relu", False),
+    ("log1p", False),
     ("round_even", False),
     ("sqrt_float", False),
     ("sqrt_int", False),

test/AotCompile/model_loader_lib.py

@@ -263,6 +263,24 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
     return TorchLoaderOutput(model=Relu(), inputs=(x,), dynamic_shapes=dynamic_shapes)
 
 
+def log1p_loader() -> TorchLoaderOutput:
+    class Log1p(torch.nn.Module):
+        def __init__(self):
+            super().__init__()
+
+        def forward(self, x: torch.Tensor) -> torch.Tensor:
+            return torch.log1p(x)
+
+    # Sample inputs
+    x = torch.randn(2, 3)
+
+    # Dynamic dim constraints
+    batch = Dim("batch")
+    dynamic_shapes = {"x": {0: batch}}
+
+    return TorchLoaderOutput(model=Log1p(), inputs=(x,), dynamic_shapes=dynamic_shapes)
+
+
 def round_even_loader() -> TorchLoaderOutput:
     class RoundEven(torch.nn.Module):
         def __init__(self):