Skip to content

Commit e1f24ec

Browse files
Arm backend: Do not partition detach copy (pytorch#18232)
Fixed an issue where _is_noop_detach_copy(node) call had been removed when check no compute partitions Signed-off-by: Christoffer J.L. <christoffer.johanssonlundqvist@arm.com>
1 parent 14ff651 commit e1f24ec

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

backends/arm/tosa/partitioner.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -260,6 +260,7 @@ def _tag_module( # noqa
260260
_is_noop_clone(node)
261261
or _is_noop_alias_copy(node)
262262
or _is_noop_expand(node)
263+
or _is_noop_detach_copy(node)
263264
or _is_noop_to_dim_order_copy(node)
264265
or _is_view_copy(node)
265266
or node.target in Q_OPS

0 commit comments

Comments
 (0)