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

Fix torch.tensor() not creating FakeTensor under FakeTensorMode

XMLWordPrintable

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

      None

      Show
      None
    • False
    • PyTorch Sprint 20

      Fixes #139092

      When torch.tensor() is called with device='meta' under FakeTensorMode, it doesn't create a FakeTensor. The function returns NotImplemented for meta devices, causing a fallback to C++ that bypasses FakeTensorMode.

              rh-ee-anandy Abhishek Nandy
              rh-ee-anandy Abhishek Nandy
              PyTorch Core
              Votes:
              0 Vote for this issue
              Watchers:
              2 Start watching this issue

                Created:
                Updated: