8000 Support boolean tensor for torch.fused_moving_avg_obs_fake_quant on CUDA by Xia-Weiwen · Pull Request #153699 · pytorch/pytorch · GitHub
[go: up one dir, main page]
Skip to content

Support boolean tensor for torch.fused_moving_avg_obs_fake_quant on CUDA #153699

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
Draft
wants to merge 3 commits into
base: main
Choose a base branch
from
Draft

Support boolean tensor for torch.fused_moving_avg_obs_fake_quant on CUDA #153699

wants to merge 3 commits into from
+14 −14

Conversation

Xia-Weiwen
Copy link
Collaborator
@Xia-Weiwen Xia-Weiwen commented May 16, 2025 •
edited
Loading

Fixes #153310

As the title

Test plan

pytest test/quantization/core/test_workflow_ops.py -k test_fused_obs_fake_quant_moving_avg

@pytorch-bot PyTorch Bot
Copy link
pytorch-bot bot commented May 16, 2025 •
edited
Loading

🔗 Helpful Links

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

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

❗ 1 Active SEVs

There are 1 currently active SEVs. If your PR is affected, please view them below:

✅ No Failures

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

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

@pytorch-bot pytorch-bot bot added the release notes: quantization release notes category label May 16, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@jerryzh168 jerryzh168 Awaiting requested review from jerryzh168 jerryzh168 will be requested when the pull request is marked ready for review jerryzh168 is a code owner

Assignees
No one assigned
Labels
open source release notes: quantization release notes category
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

IInconsistent Error Handling in torch.fused_moving_avg_obs_fake_quant Between CPU and GPU Implementations
2 participants
@Xia-Weiwen @pytorchbot
0