Skip to content

DTensor redistribuite is not supported from thunderfx if placements is provided as positional arg #2794

@mattteochen

Description

@mattteochen

Currently thunder supports:

z = z.redistribute(placements=[Replicate()], async_op=True)

This will mark the redistribution node as unsupported:

z = z.redistribute(mesh, [Replicate()], async_op=True)

The issue should be this dictionary access, while placements could be passed as a positional argument.

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions