Skip to content

[Bug] Unsupported function types ['fmod.Tensor'] #17880

@syheliel

Description

@syheliel

This bug is found by fuzzing, and there are also some other similar bug related to unsupported function types, Is there a better way to report them?

Environment

  • tvm 0.21.dev26+g2ca6ec8a5
  • torch 2.6.0

Steps to reproduce

import torch
from tvm.relax import frontend
from tvm.relax.frontend.torch import from_exported_program
from torch.export import export
import torch.nn as nn

class FModModule(nn.Module):
    def __init__(self):
        super(FModModule, self).__init__()
    
    def forward(self, x, y):
        return torch.fmod(x, y)

if __name__ == "__main__":
    model = FModModule()
    example_args = (torch.randn(3, 3), torch.randn(3, 3))
    
    with torch.no_grad():
        exported_program = export(model, example_args)
        mod = from_exported_program(exported_program, keep_params_as_input=True)
    
    mod, params = frontend.detach_params(mod)
    mod.show()

Triage

  • frontend:pytorch

Metadata

Metadata

Assignees

No one assigned

    Labels

    needs-triagePRs or issues that need to be investigated by maintainers to find the right assignees to address ittype: bug

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions