Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Nov 13, 2024
1 parent 4f548eb commit 9f629cf
Showing 1 changed file with 8 additions and 4 deletions.
12 changes: 8 additions & 4 deletions thunder/executors/nvfuserex_impl.py
Original file line number Diff line number Diff line change
Expand Up @@ -441,8 +441,8 @@ class FusionDefinitionWrapper:
cache_clear: None | Callable = None
last_used: None | FusionDefinition = None
last_inputs: None | Sequence[tuple] = None
store_inputs: bool = False,
_enable_options: None | list[str] = None,
store_inputs: bool = (False,)
_enable_options: None | list[str] = (None,)
_disable_options: None | list[str] = None

def __call__(self, *args):
Expand Down Expand Up @@ -559,8 +559,12 @@ def create_fusion_definition_wrapper(
store_inputs: None | bool = get_compile_option(
"nv_store_fusion_inputs", "Allow nvFuser to store fusion inputs for repro."
)
_enable_options: None | list[str] = get_compile_option("nv_enable_options", "List of NVFUSER_ENABLE options to set.")
_disable_options: None | list[str] = get_compile_option("nv_disable_options", "List of NVFUSER_DISABLE options to set.")
_enable_options: None | list[str] = get_compile_option(
"nv_enable_options", "List of NVFUSER_ENABLE options to set."
)
_disable_options: None | list[str] = get_compile_option(
"nv_disable_options", "List of NVFUSER_DISABLE options to set."
)

tensor_indices = []
for idx, x in enumerate(sorted_unique_inputs):
Expand Down

0 comments on commit 9f629cf

Please sign in to comment.