Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[Torch][Graph][Operator] Add and fix various items for torchvision model support #347

Merged
merged 14 commits into from Aug 12, 2023
Merged

[Torch][Graph][Operator] Add and fix various items for torchvision model support #347

merged 14 commits into from Aug 12, 2023

Conversation

hjjq
Copy link
Member

@hjjq hjjq commented Aug 11, 2023 •
edited

  1. Enhance support for __setitem__ and __getitem__ of Tensor; Add SetStridedSlice Op, Roll Op.
  2. Add/Update torch mapping for adaptive_avg_pool3d, eq, pad, roll, matmul, new_zeros, batch_norm, MultiHeadAttention.
  3. Update torch Linear mapping to optionally accept transposed weights.
  4. Fix a bug where a empty graph will output a zero tensor instead of the input/weight.

yaoyaoding
yaoyaoding approved these changes Aug 12, 2023
View reviewed changes
Copy link
Member

@yaoyaoding yaoyaoding left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

It looks good to me, good job!

@hjjq hjjq merged commit 740ff3c into hidet-org:main Aug 12, 2023
2 checks passed
@hjjq hjjq mentioned this pull request Aug 15, 2023
Merged
hjjq added a commit that referenced this pull request Aug 15, 2023
@hjjq
[Torch][Operator] More torchvision model support (#348)
1fb45f7 
This is a continuation of #347.

1. Add LP normalization task (ToDo: schedule template)
2. Add torch mappings for normalize, clone, zero_, exp, chunk
3. Add ceil_mode=True support for pool2d
4. Fix dtype issue in resize
5. Fix other bugs in pad, conv2d_pattern
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@yaoyaoding yaoyaoding yaoyaoding approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

None yet
2 participants
@hjjq @yaoyaoding