Uploaded image for project: 'AI Platform Core Components'
  1. AI Platform Core Components
  2. AIPCC-5715

[feature][dtensor] support nn.Module inputs containing DTensor parameters in local_map #163692

XMLWordPrintable

    • Icon: Story Story
    • Resolution: Unresolved
    • Icon: Undefined Undefined
    • None
    • None
    • PyTorch
    • None
    • False
    • Hide

      None

      Show
      None
    • False
    • PyTorch Sprint 17

      for more details please have a look into this upstream issue. Thanks!

      https://github.com/pytorch/pytorch/issues/163692

              rh-ee-rrathaur Rohit Singh Rathaur
              rh-ee-rrathaur Rohit Singh Rathaur
              PyTorch Distributed
              Votes:
              0 Vote for this issue
              Watchers:
              3 Start watching this issue

                Created:
                Updated: