Skip to content

[inductor] Add bitwise tests for compiled Adam/AdamW#174911

Closed
mlazos wants to merge 40 commits intogh/mlazos/100/basefrom
gh/mlazos/100/head
Closed

[inductor] Add bitwise tests for compiled Adam/AdamW#174911
mlazos wants to merge 40 commits intogh/mlazos/100/basefrom
gh/mlazos/100/head

Conversation

@mlazos
Copy link
Copy Markdown
Contributor

@mlazos mlazos commented Feb 12, 2026

Stack from ghstack (oldest at bottom):

Add CompiledOptimizerBitwiseTests test suite that verifies compiled
optimizers produce bitwise identical results to eager when precision
configs are enabled:

  • eager_numerics.division_rounding = True
  • eager_numerics.pow_precision = True
  • emulate_precision_casts = True

Tests cover Adam and AdamW with various configurations including
amsgrad, maximize, and weight_decay options.

cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @ipiszy @kadeng @muchulee8 @amjames @chauhang @aakhundov @coconutruben @jataylo

[ghstack-poisoned]
@pytorch-bot
Copy link
Copy Markdown

pytorch-bot Bot commented Feb 12, 2026

🔗 Helpful Links

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

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

✅ You can merge normally! (3 Unrelated Failures)

As of commit 2dda222 with merge base a6beff3 (image):

FLAKY - The following job failed but was likely due to flakiness present on trunk:

BROKEN TRUNK - The following job failed but was present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

UNSTABLE - The following job is marked as unstable, possibly due to flakiness on trunk:

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

mlazos added a commit that referenced this pull request Feb 12, 2026
Add CompiledOptimizerBitwiseTests test suite that verifies compiled
optimizers produce bitwise identical results to eager when precision
configs are enabled:
- eager_numerics.division_rounding = True
- eager_numerics.pow_precision = True
- emulate_precision_casts = True

Tests cover Adam and AdamW with various configurations including
amsgrad, maximize, and weight_decay options.


ghstack-source-id: a1c6174
Pull-Request: #174911
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
mlazos added a commit that referenced this pull request Feb 18, 2026
Add CompiledOptimizerBitwiseTests test suite that verifies compiled
optimizers produce bitwise identical results to eager when precision
configs are enabled:
- eager_numerics.division_rounding = True
- eager_numerics.pow_precision = True
- emulate_precision_casts = True

Tests cover Adam and AdamW with various configurations including
amsgrad, maximize, and weight_decay options.


ghstack-source-id: 95f8455
Pull-Request: #174911
[ghstack-poisoned]
mlazos added a commit that referenced this pull request Feb 18, 2026
Add CompiledOptimizerBitwiseTests test suite that verifies compiled
optimizers produce bitwise identical results to eager when precision
configs are enabled:
- eager_numerics.division_rounding = True
- eager_numerics.pow_precision = True
- emulate_precision_casts = True

Tests cover Adam and AdamW with various configurations including
amsgrad, maximize, and weight_decay options.


ghstack-source-id: 2460231
Pull-Request: #174911
mlazos added a commit that referenced this pull request Feb 18, 2026
Add CompiledOptimizerBitwiseTests test suite that verifies compiled
optimizers produce bitwise identical results to eager when precision
configs are enabled:
- eager_numerics.division_rounding = True
- eager_numerics.pow_precision = True
- emulate_precision_casts = True

Tests cover Adam and AdamW with various configurations including
amsgrad, maximize, and weight_decay options.


ghstack-source-id: 2460231
Pull-Request: #174911
[ghstack-poisoned]
mlazos added a commit that referenced this pull request Feb 18, 2026
Add CompiledOptimizerBitwiseTests test suite that verifies compiled
optimizers produce bitwise identical results to eager when precision
configs are enabled:
- eager_numerics.division_rounding = True
- eager_numerics.pow_precision = True
- emulate_precision_casts = True

Tests cover Adam and AdamW with various configurations including
amsgrad, maximize, and weight_decay options.


ghstack-source-id: c708d67
Pull-Request: #174911
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
mlazos added a commit that referenced this pull request Feb 19, 2026
Add CompiledOptimizerBitwiseTests test suite that verifies compiled
optimizers produce bitwise identical results to eager when precision
configs are enabled:
- eager_numerics.division_rounding = True
- eager_numerics.pow_precision = True
- emulate_precision_casts = True

Tests cover Adam and AdamW with various configurations including
amsgrad, maximize, and weight_decay options.

ghstack-source-id: 2e46fb4
Pull-Request: #174911
[ghstack-poisoned]
mlazos added a commit that referenced this pull request Feb 28, 2026
Add CompiledOptimizerBitwiseTests test suite that verifies compiled
optimizers produce bitwise identical results to eager when precision
configs are enabled:
- eager_numerics.division_rounding = True
- eager_numerics.pow_precision = True
- emulate_precision_casts = True

Tests cover Adam and AdamW with various configurations including
amsgrad, maximize, and weight_decay options.

ghstack-source-id: ee65756
Pull-Request: #174911
[ghstack-poisoned]
mlazos added a commit that referenced this pull request Feb 28, 2026
Add CompiledOptimizerBitwiseTests test suite that verifies compiled
optimizers produce bitwise identical results to eager when precision
configs are enabled:
- eager_numerics.division_rounding = True
- eager_numerics.pow_precision = True
- emulate_precision_casts = True

Tests cover Adam and AdamW with various configurations including
amsgrad, maximize, and weight_decay options.

ghstack-source-id: 28662ba
Pull-Request: #174911
[ghstack-poisoned]
mlazos added a commit that referenced this pull request Mar 2, 2026
Add CompiledOptimizerBitwiseTests test suite that verifies compiled
optimizers produce bitwise identical results to eager when precision
configs are enabled:
- eager_numerics.division_rounding = True
- eager_numerics.pow_precision = True
- emulate_precision_casts = True

Tests cover Adam and AdamW with various configurations including
amsgrad, maximize, and weight_decay options.

ghstack-source-id: 9032d6c
Pull-Request: #174911
[ghstack-poisoned]
mlazos added a commit that referenced this pull request Mar 2, 2026
Add CompiledOptimizerBitwiseTests test suite that verifies compiled
optimizers produce bitwise identical results to eager when precision
configs are enabled:
- eager_numerics.division_rounding = True
- eager_numerics.pow_precision = True
- emulate_precision_casts = True

Tests cover Adam and AdamW with various configurations including
amsgrad, maximize, and weight_decay options.

ghstack-source-id: 13a2da2
Pull-Request: #174911
mlazos added a commit that referenced this pull request Mar 2, 2026
Add CompiledOptimizerBitwiseTests test suite that verifies compiled
optimizers produce bitwise identical results to eager when precision
configs are enabled:
- eager_numerics.division_rounding = True
- eager_numerics.pow_precision = True
- emulate_precision_casts = True

Tests cover Adam and AdamW with various configurations including
amsgrad, maximize, and weight_decay options.

ghstack-source-id: 13a2da2
Pull-Request: #174911
[ghstack-poisoned]
mlazos added a commit that referenced this pull request Mar 2, 2026
Add CompiledOptimizerBitwiseTests test suite that verifies compiled
optimizers produce bitwise identical results to eager when precision
configs are enabled:
- eager_numerics.division_rounding = True
- eager_numerics.pow_precision = True
- emulate_precision_casts = True

Tests cover Adam and AdamW with various configurations including
amsgrad, maximize, and weight_decay options.

ghstack-source-id: 1b160e5
Pull-Request: #174911
[ghstack-poisoned]
mlazos added a commit that referenced this pull request Mar 2, 2026
Add CompiledOptimizerBitwiseTests test suite that verifies compiled
optimizers produce bitwise identical results to eager when precision
configs are enabled:
- eager_numerics.division_rounding = True
- eager_numerics.pow_precision = True
- emulate_precision_casts = True

Tests cover Adam and AdamW with various configurations including
amsgrad, maximize, and weight_decay options.

ghstack-source-id: 7ddcd24
Pull-Request: #174911
[ghstack-poisoned]
mlazos added a commit that referenced this pull request Mar 2, 2026
Add CompiledOptimizerBitwiseTests test suite that verifies compiled
optimizers produce bitwise identical results to eager when precision
configs are enabled:
- eager_numerics.division_rounding = True
- eager_numerics.pow_precision = True
- emulate_precision_casts = True

Tests cover Adam and AdamW with various configurations including
amsgrad, maximize, and weight_decay options.

ghstack-source-id: a7031f2
Pull-Request: #174911
[ghstack-poisoned]
mlazos added a commit that referenced this pull request Mar 3, 2026
Add CompiledOptimizerBitwiseTests test suite that verifies compiled
optimizers produce bitwise identical results to eager when precision
configs are enabled:
- eager_numerics.division_rounding = True
- eager_numerics.pow_precision = True
- emulate_precision_casts = True

Tests cover Adam and AdamW with various configurations including
amsgrad, maximize, and weight_decay options.

ghstack-source-id: b59afbb
Pull-Request: #174911
@mlazos
Copy link
Copy Markdown
Contributor Author

mlazos commented Mar 3, 2026

@pytorchbot merge

@pytorch-bot pytorch-bot Bot added the ciflow/trunk Trigger trunk jobs on your pull request label Mar 3, 2026
@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

postmath pushed a commit to postmath/pytorch that referenced this pull request Mar 3, 2026
Add CompiledOptimizerBitwiseTests test suite that verifies compiled
optimizers produce bitwise identical results to eager when precision
configs are enabled:
- eager_numerics.division_rounding = True
- eager_numerics.pow_precision = True
- emulate_precision_casts = True

Tests cover Adam and AdamW with various configurations including
amsgrad, maximize, and weight_decay options.

Pull Request resolved: pytorch#174911
Approved by: https://github.com/v0i0
ghstack dependencies: pytorch#176237, pytorch#175839
EmanueleCoradin pushed a commit to EmanueleCoradin/pytorch that referenced this pull request Mar 30, 2026
Add CompiledOptimizerBitwiseTests test suite that verifies compiled
optimizers produce bitwise identical results to eager when precision
configs are enabled:
- eager_numerics.division_rounding = True
- eager_numerics.pow_precision = True
- emulate_precision_casts = True

Tests cover Adam and AdamW with various configurations including
amsgrad, maximize, and weight_decay options.

Pull Request resolved: pytorch#174911
Approved by: https://github.com/v0i0
ghstack dependencies: pytorch#176237, pytorch#175839
@github-actions github-actions Bot deleted the gh/mlazos/100/head branch April 3, 2026 02:26
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants