lint

Author: Gasoonjia

backends/aoti/aoti_backend.py

@@ -9,7 +9,7 @@
 import typing
 from abc import ABC, abstractmethod
 from enum import Enum
-from typing import Any, Dict, List, Set
+from typing import Any, Dict, List, Optional, Set
 
 import torch
 from executorch.backends.aoti.passes.replace_view_copy_with_view import (
@@ -88,7 +88,9 @@ def save_data_externally(cls) -> bool:
         return False
 
     @classmethod
-    def get_extra_aoti_compile_context_manager(cls, compile_specs: List[CompileSpec]):
+    def get_extra_aoti_compile_context_manager(
+        cls, compile_specs: Optional[List[CompileSpec]] = None
+    ):
         """Return extra context manager to apply during aoti_compile stage. By default returns an empty context manager.
 
         Subclasses may inspect ``compile_specs`` to opt into behaviors that

backends/cuda/cuda_backend.py

@@ -332,7 +332,9 @@ def get_aoti_compile_options(
         return options
 
     @classmethod
-    def get_extra_aoti_compile_context_manager(cls, compile_specs: List[CompileSpec]):
+    def get_extra_aoti_compile_context_manager(
+        cls, compile_specs: Optional[List[CompileSpec]] = None
+    ):
         """
         Combine all extra context managers needed during AOTInductor
         compilation for the CUDA backend. Each manager is documented at
@@ -344,9 +346,10 @@ def get_extra_aoti_compile_context_manager(cls, compile_specs: List[CompileSpec]
         through the unmodified AOTI codepath, which avoids regressions in
         their cuda CI exports.
         """
-        # Parse compile_specs for low_memory_mode (default OFF)
+        # Parse compile_specs for low_memory_mode (default OFF). compile_specs
+        # may be None when called without specs (parity with base default).
         low_memory_mode = "OFF"
-        for spec in compile_specs:
+        for spec in compile_specs or []:
             if spec.key == "low_memory_mode":
                 mode = spec.value.decode("utf-8").upper()
                 if mode not in ["ON", "OFF"]: