Skip to content

Commit 45dffd6

Browse files
committed
Add impl for aten.round.
1 parent fc512b6 commit 45dffd6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/complex_tensor/ops/aten.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,6 @@ def is_pinned_impl(self: ComplexTensor, device: torch.device | None = None) -> b
7070
unsqueeze_impl = register_simple(aten.unsqueeze)
7171
mean_impl = register_simple(aten.mean)
7272
sum_impl = register_simple(aten.sum)
73-
round_impl = register_simple(aten.round)
7473
clone_impl = register_simple(aten.clone)
7574
neg_impl = register_simple(aten.neg)
7675
flip_impl = register_simple(aten.flip)
@@ -318,6 +317,7 @@ def isclose_impl(
318317
aten.argmin,
319318
aten.sort,
320319
aten.topk,
320+
aten.round,
321321
]
322322

323323

0 commit comments

Comments
 (0)