Skip to content

Commit 40de63b

Browse files
zero000064pytorchmergebot
authored andcommitted
parameterized test_graph_optims and test_graph_scaling_fused_optimizers (#133749)
Fixes #123451 This is a rework of a reverted pull request, #125127. The test failure is fixed. Pull Request resolved: #133749 Approved by: https://github.com/janeyx99
1 parent c7338f4 commit 40de63b

File tree

3 files changed

+230
-217
lines changed

3 files changed

+230
-217
lines changed

test/inductor/test_compiled_optimizers.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -146,6 +146,8 @@ class KernelCounts(NamedTuple):
146146
"test_sgd_weight_decay_maximize_cuda": 4,
147147
"test_sgd_weight_decay_maximize_xpu": 4,
148148
"test_sgd_weight_decay_maximize_cpu": 4,
149+
"test_sgd_weight_decay_cpu": 4,
150+
"test_sgd_weight_decay_cuda": 4,
149151
"test_sgd_momentum_weight_decay_foreach_cuda": 2,
150152
"test_sgd_momentum_weight_decay_foreach_xpu": 2,
151153
"test_sgd_momentum_nesterov_weight_decay_foreach_cuda": 2,

0 commit comments

Comments
 (0)