已合并
[v2.10.0] cherry-pick v2.7.1_fix_ut: remove unskip #37617
ffmh创建于 6月4日
[v2.10.0] cherry-pick v2.7.1_fix_ut: remove unskip #37617
已合并
共 1 个文件变更+0-3
| @@ -138,7 +138,6 @@ class TestFusedOptim(TestCase): | |||
| 138 | if p.grad is not None: | 138 | if p.grad is not None: |
| 139 | self.assertRtolEqual(p.sum().item(), self.third_optim_baseline[fused_opt_obj][i]) | 139 | self.assertRtolEqual(p.sum().item(), self.third_optim_baseline[fused_opt_obj][i]) |
| 140 | 140 | ||
| 141 | - | ||
| 142 | def test_unscale(self): | 141 | def test_unscale(self): |
| 143 | model = self._create_simple_model() | 142 | model = self._create_simple_model() |
| 144 | input_tensor = torch.rand(3, 1, 24, 24).to("npu:0") | 143 | input_tensor = torch.rand(3, 1, 24, 24).to("npu:0") |
| @@ -161,7 +160,6 @@ class TestFusedOptim(TestCase): | |||
| 161 | if p.grad is not None: | 160 | if p.grad is not None: |
| 162 | self.assertEqual(grads_before_unscale[p] / 128, p.grad) | 161 | self.assertEqual(grads_before_unscale[p] / 128, p.grad) |
| 163 | 162 | ||
| 164 | - | ||
| 165 | def test_simple_model_train_dynamic(self): | 163 | def test_simple_model_train_dynamic(self): |
| 166 | model = self._create_simple_model() | 164 | model = self._create_simple_model() |
| 167 | optim_cases = self._create_optimizer_cases() | 165 | optim_cases = self._create_optimizer_cases() |
| @@ -192,7 +190,6 @@ class TestFusedOptim(TestCase): | |||
| 192 | scaler_fused.update() | 190 | scaler_fused.update() |
| 193 | self.assertRtolEqual(loss, loss_fused) | 191 | self.assertRtolEqual(loss, loss_fused) |
| 194 | 192 | ||
| 195 | - | ||
| 196 | def test_simple_model_train_static(self): | 193 | def test_simple_model_train_static(self): |
| 197 | model = self._create_simple_model() | 194 | model = self._create_simple_model() |
| 198 | optim_cases = self._create_optimizer_cases() | 195 | optim_cases = self._create_optimizer_cases() |