Skip to content

Commit f2af0f1

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 0b5b584 commit f2af0f1

File tree

1 file changed

+1
-0
lines changed
  • transformer_engine/jax/cpp_extensions

1 file changed

+1
-0
lines changed

transformer_engine/jax/cpp_extensions/base.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -289,6 +289,7 @@ def _parse_custom_call_string_to_dict(custom_calls_str: str):
289289
settings[key] = value == "true"
290290
return settings
291291

292+
292293
@contextmanager
293294
def primitive_context(primitive_enabling_changes: str):
294295
orig_env_var = os.getenv("NVTE_JAX_CUSTOM_CALLS")

0 commit comments

Comments
 (0)