Skip to content

Patch the _is_conv_node function#153749

Closed
cccclai wants to merge 1 commit into
pytorch:mainfrom
cccclai:export-D74898941
Closed

Patch the _is_conv_node function#153749
cccclai wants to merge 1 commit into
pytorch:mainfrom
cccclai:export-D74898941

Conversation

@cccclai

@cccclai cccclai commented May 16, 2025

Copy link
Copy Markdown
Contributor

Summary: torch.ops.aten.conv2d.padding is also conv2d node

Differential Revision: D74898941

@pytorch-bot

pytorch-bot Bot commented May 16, 2025

Copy link
Copy Markdown

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/153749

Note: Links to docs will display an error until the docs builds have been completed.

⏳ No Failures, 47 Pending

As of commit 725c2f6 with merge base 4277907 (image):
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@facebook-github-bot

Copy link
Copy Markdown
Contributor

This pull request was exported from Phabricator. Differential Revision: D74898941

@cccclai cccclai requested review from andrewor14 and jerryzh168 May 16, 2025 19:03
@pytorch-bot pytorch-bot Bot added the ciflow/trunk Trigger trunk jobs on your pull request label May 19, 2025
Comment thread torch/ao/quantization/pt2e/utils.py Outdated
Comment on lines 169 to 172

@Skylion007 Skylion007 May 19, 2025

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

Suggested change
torch.ops.aten.conv1d.default,
torch.ops.aten.conv2d.default,
torch.ops.aten.conv2d.padding,
torch.ops.aten.conv1d.default,
torch.ops.aten.conv1d.padding,
torch.ops.aten.conv2d.default,
torch.ops.aten.conv2d.padding,

What about conv3d?

Suggested change
torch.ops.aten.conv1d.default,
torch.ops.aten.conv2d.default,
torch.ops.aten.conv2d.padding,
torch.ops.aten.conv1d.default,
torch.ops.aten.conv1d.padding,
torch.ops.aten.conv2d.default,
torch.ops.aten.conv2d.padding,
torch.ops.aten.conv3d.default,
torch.ops.aten.conv3d.padding,

@jerryzh168 jerryzh168 May 19, 2025

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

I added conv3d in torchao/quantization/pt2e,

@cccclai might be good to abandon these changes

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

Wait so this func is unused now?

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

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

Our partner is still using pytorch but not torch ao currently, I think it's an easy fix?

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

@cccclai Might as well be complete and ad all the conv types then.

@Skylion007

Copy link
Copy Markdown
Collaborator

Do we need to add any tests for the quantization ops?

cccclai added a commit to cccclai/ao that referenced this pull request May 19, 2025
Summary:
X-link: pytorch/pytorch#153749

torch.ops.aten.conv2d.padding is also conv2d node

Reviewed By: andrewor14, jerryzh168

Differential Revision: D74898941

@Skylion007 Skylion007 left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

LGTM if you also fix the utility here

@cccclai cccclai force-pushed the export-D74898941 branch from 110aee3 to 7a144ea Compare May 20, 2025 05:54
pytorch-bot Bot pushed a commit that referenced this pull request May 20, 2025
Summary:
X-link: pytorch/ao#2223


torch.ops.aten.conv2d.padding is also conv2d node

Test Plan: CI

Reviewed By: andrewor14, jerryzh168

Differential Revision: D74898941
@facebook-github-bot

Copy link
Copy Markdown
Contributor

This pull request was exported from Phabricator. Differential Revision: D74898941

@cccclai

cccclai commented May 20, 2025

Copy link
Copy Markdown
Contributor Author

LGTM if you also fix the utility here

Does it look good now?

@Skylion007

Copy link
Copy Markdown
Collaborator

@cccclai Yeah, just fix the trailing spaces in the lint

cccclai added a commit to cccclai/ao that referenced this pull request May 20, 2025
Summary:
Pull Request resolved: pytorch#2223

X-link: pytorch/pytorch#153749

torch.ops.aten.conv2d.padding is also conv2d node

Reviewed By: andrewor14, jerryzh168

Differential Revision: D74898941
cccclai added a commit to cccclai/ao that referenced this pull request May 22, 2025
Summary:

X-link: pytorch/pytorch#153749

torch.ops.aten.conv2d.padding is also conv2d node

Reviewed By: andrewor14, jerryzh168

Differential Revision: D74898941
@cccclai cccclai force-pushed the export-D74898941 branch from 7a144ea to f7a791e Compare May 22, 2025 18:13
@facebook-github-bot

Copy link
Copy Markdown
Contributor

This pull request was exported from Phabricator. Differential Revision: D74898941

@cccclai

cccclai commented May 22, 2025

Copy link
Copy Markdown
Contributor Author

I added the test in pytorch/ao#2223. The same test doesn't work in pytorch, looks like the quant nodes disappear.

cccclai added a commit to cccclai/ao that referenced this pull request May 22, 2025
Summary:

X-link: pytorch/pytorch#153749

torch.ops.aten.conv2d.padding is also conv2d node

Reviewed By: andrewor14, jerryzh168

Differential Revision: D74898941
Summary:
X-link: pytorch/ao#2223


torch.ops.aten.conv2d.padding is also conv2d node

Test Plan: CI

Reviewed By: andrewor14, jerryzh168

Differential Revision: D74898941
@cccclai cccclai force-pushed the export-D74898941 branch from f7a791e to 725c2f6 Compare May 22, 2025 18:59
@facebook-github-bot

Copy link
Copy Markdown
Contributor

This pull request was exported from Phabricator. Differential Revision: D74898941

@facebook-github-bot

Copy link
Copy Markdown
Contributor

@pytorchbot merge

(Initiating merge automatically since Phabricator Diff has merged)

@pytorchmergebot

Copy link
Copy Markdown
Collaborator

Merge started

Your change will be merged once all checks pass (ETA 0-4 Hours).

Learn more about merging in the wiki.

Questions? Feedback? Please reach out to the PyTorch DevX Team

Advanced Debugging
Check the merge workflow status
here

@facebook-github-bot

Copy link
Copy Markdown
Contributor

@pytorchbot revert -m="Diff reverted internally" -c="ghfirst"

This Pull Request has been reverted by a revert inside Meta. To re-land this change, please open another pull request, assign the same reviewers, fix the CI failures that caused the revert and make sure that the failing CI runs on the PR by applying the proper ciflow label (e.g., ciflow/trunk).)

@pytorchmergebot

Copy link
Copy Markdown
Collaborator

@pytorchbot successfully started a revert job. Check the current status here.
Questions? Feedback? Please reach out to the PyTorch DevX Team

pytorchmergebot added a commit that referenced this pull request May 23, 2025
This reverts commit c985cec.

Reverted #153749 on behalf of https://github.com/facebook-github-bot due to Diff reverted internally ([comment](#153749 (comment)))
@pytorchmergebot

Copy link
Copy Markdown
Collaborator

@cccclai your PR has been successfully reverted.

@pytorchmergebot pytorchmergebot added Reverted ci-no-td Do not run TD on this PR labels May 23, 2025
@cccclai

cccclai commented May 23, 2025

Copy link
Copy Markdown
Contributor Author

Close as the PR is reverted

@cccclai cccclai closed this May 23, 2025
@cccclai

cccclai commented May 28, 2025

Copy link
Copy Markdown
Contributor Author

Reland the pr in #154473 and pytorch/ao#2257

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ci-no-td Do not run TD on this PR ciflow/trunk Trigger trunk jobs on your pull request fb-exported Merged release notes: AO frontend release notes: quantization release notes category Reverted

Projects

None yet

Development

Successfully merging this pull request may close these issues.

6 participants