Skip to content
This repository was archived by the owner on Aug 1, 2025. It is now read-only.
This repository was archived by the owner on Aug 1, 2025. It is now read-only.

[inductor] Fallback can't handle ops with output signature (Tensor, Tensor[]) #1805

@anijain2305

Description

@anijain2305

🐛 Describe the bug

Repro

import torch
import torch._dynamo as dynamo

def fn():
    return torch.histogramdd(torch.tensor([[0., 1.], [1., 0.], [2., 0.], [2., 2.]]), bins=[3, 3],
                  weight=torch.tensor([1., 2., 4., 8.]))

print(fn())

opt_fn = dynamo.optimize()(fn)
opt_fn()

Error logs

Traceback (most recent call last):
  File "/scratch/anijain/work/pytorch/torch/_inductor/graph.py", line 239, in call_function
    out = lowerings[target](*args, **kwargs)
  File "/scratch/anijain/work/pytorch/torch/_inductor/lowering.py", line 168, in wrapped
    assert not any(isinstance(x, TensorBox) for x in kwargs.values())
AssertionError

Minified repro

Already minified

Metadata

Metadata

Assignees

Labels

bugSomething isn't workingtriaged

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions