Skip to content

Commit f8dd3ad

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

File tree

2 files changed

+3
-5
lines changed

2 files changed

+3
-5
lines changed

thunder/executors/nvfuserex_impl.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@
1717
if IS_TORCH_DISTRIBUTED_AVAILABLE:
1818
from torch.distributed.tensor import DTensor
1919
from torch.distributed.tensor.placement_types import Placement, Shard
20-
import torch.distributed as dist
2120

2221
import thunder.core.dtypes as dtypes
2322
import thunder.torch as ltorch

thunder/tests/distributed/test_dtensor.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -458,7 +458,6 @@ def test_dtensor_opinfo(self, op: OpInfo, executor):
458458

459459
assert tested_sample_count > 0, f"test_dtensor_opinfo:No samples tested for {op.name} with {executor} executor"
460460

461-
462461
@common_utils.parametrize("executor", tuple(executors_map.keys()))
463462
def test_use_dtensor_execute_flag_with_dtensor(self, executor):
464463
"""Test that use_dtensor_execute flag is set to True for DTensor inputs."""
@@ -494,9 +493,9 @@ def fn(x, w):
494493
fusion_name = fusion_bsym.sym.name
495494
fdw = call_ctx.get(fusion_name)
496495
assert fdw is not None, f"FusionDefinitionWrapper not found for {fusion_name}"
497-
assert hasattr(
498-
fdw, "use_dtensor_execute"
499-
), "FusionDefinitionWrapper should have use_dtensor_execute attribute"
496+
assert hasattr(fdw, "use_dtensor_execute"), (
497+
"FusionDefinitionWrapper should have use_dtensor_execute attribute"
498+
)
500499
assert fdw.use_dtensor_execute is True, "use_dtensor_execute should be True for DTensor inputs"
501500

502501
# Verify the result is correct

0 commit comments

Comments
 (0)