Skip to content

[Relax][PyTorch] Support tensor manipulation and creation ops for ExportedProgram importer#17429

Merged
mshr-h merged 20 commits into
apache:mainfrom
mshr-h:relax-exportedprogram-10
Oct 2, 2024
Merged

[Relax][PyTorch] Support tensor manipulation and creation ops for ExportedProgram importer#17429
mshr-h merged 20 commits into
apache:mainfrom
mshr-h:relax-exportedprogram-10

Conversation

@mshr-h

@mshr-h mshr-h commented Sep 30, 2024

Copy link
Copy Markdown
Contributor

Part of #17346

With this PR, I verified that the following torchvision models can be executed.

  • convnext_tiny
  • densenet121
  • efficientnet_b0
  • efficientnet_v2_s
  • inception_v3
  • maxvit_t
  • mnasnet0_5
  • mobilenet_v2
  • mobilenet_v3_small
  • regnet_x_400mf
  • resnet18
  • resnext50_32x4d
  • shufflenet_v2_x0_5
  • squeezenet1_0
  • swin_t
  • swin_v2_t
  • vgg11
  • vgg11_bn
  • vit_b_32
  • wide_resnet50_2

@mshr-h

mshr-h commented Sep 30, 2024

Copy link
Copy Markdown
Contributor Author

@tvm-bot rerun

@mshr-h

mshr-h commented Oct 1, 2024

Copy link
Copy Markdown
Contributor Author

@tvm-bot rerun

@mshr-h

mshr-h commented Oct 1, 2024

Copy link
Copy Markdown
Contributor Author

The hexagon test is failing, but I don't know why.
Seems like all the tests are passing but the status shows it's failed.
cc @tqchen

@tqchen

tqchen commented Oct 1, 2024

Copy link
Copy Markdown
Member

looks like timeout, maybe we could move some of the integration tests to nightly

@tqchen

tqchen commented Oct 1, 2024

Copy link
Copy Markdown
Member

@tvm-bot rerun

@mshr-h mshr-h merged commit fab67a9 into apache:main Oct 2, 2024
@mshr-h mshr-h deleted the relax-exportedprogram-10 branch October 2, 2024 01:57
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants