increase profiler timeout to 500

facebookincubator · Jul 30, 2023 · b324f54 · b324f54
1 parent 1440635
commit b324f54
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/python/aitemplate/backend/profiler_runner.py b/python/aitemplate/backend/profiler_runner.py
@@ -256,7 +256,7 @@ class ProfilerRunner:
     however, the results are empirically better compared to the previous runner.
     """
 
-    def __init__(self, devices: List[str], postprocessing_delegate, timeout: int = 300):
+    def __init__(self, devices: List[str], postprocessing_delegate, timeout: int = 500):
         """
         Parameters
         ----------

diff --git a/python/aitemplate/compiler/compiler.py b/python/aitemplate/compiler/compiler.py
@@ -160,7 +160,7 @@ def compile_model(
     allocator_kind: Optional[AITemplateAllocatorKind] = None,
     debug_settings: AITDebugSettings = _DEBUG_SETTINGS,
     do_optimize_graph: bool = True,
-    profile_timeout: int = 300,
+    profile_timeout: int = 500,
 ) -> Model:
     """Compiles a model and generates a .so file.
 

diff --git a/python/aitemplate/compiler/transform/profile.py b/python/aitemplate/compiler/transform/profile.py
@@ -56,7 +56,7 @@ def profile(
     workdir="./tmp",
     devices=None,
     dynamic_profiling_strategy=DynamicProfileStrategy.MAX,
-    timeout=300,
+    timeout=500,
 ):
     """Profiles kernels.