Optional get_example_inputs method of model

README.md

@@ -298,6 +298,13 @@ class Model(torch.nn.Module):
             num_warps=32,
         )
         return OUT
+
+      # Optional
+      # If this method is provided, the Xe-Forge will get input tensors by this method.
+      # If not, Xe-Forge will generate random inputs based on shapes and dtype.
+      # Don't provide this method (remove it) if you don't need it.
+      def get_example_inputs(self, input_shapes: list | None = None):
+        pass
 ```
 
 ### Model with Init Arguments

src/xe_forge/core/executor.py

@@ -153,7 +153,9 @@ def execute(
 
             # Create inputs if not provided
             if inputs is None:
-                if input_shapes:
+                if hasattr(model, "get_example_inputs"):
+                    inputs = model.get_example_inputs(input_shapes, self.device)
+                elif input_shapes:
                     inputs = self._create_inputs(
                         input_shapes, dtype=dtype, input_dtypes=input_dtypes
                     )
@@ -302,7 +304,10 @@ def _check_correctness(
 
             # Shared inputs with deterministic seed
             set_all_seeds(123)
-            inputs = self._create_inputs(input_shapes, dtype=dtype, input_dtypes=input_dtypes)
+            if hasattr(original_model, "get_example_inputs"):
+                inputs = original_model.get_example_inputs(input_shapes, self.device)
+            else:
+                inputs = self._create_inputs(input_shapes, dtype=dtype, input_dtypes=input_dtypes)
 
             inputs_orig = [inp.clone() for inp in inputs]
             inputs_opt = [inp.clone() for inp in inputs]