Platforms: linux

This test was disabled because it is failing on master ([recent examples](http://torch-ci.com/failure/test_torch.py%3A%3ATestTorch%3A%3Atest_equal)).

The cause seems to point to https://github.com/pytorch/pytorch/pull/100024 (or D45282119 as it was landed internally)

cc @mruberry @houseroad @ezyang 
a.scale(torch.tensor([4.0], d=torch.float32, d=d))
                self.assertTrue(isinstance(a._scale, torch.cuda.FloatTensor))
            
            serialized = pickle.dumps(a)
            b = pickle.loads(serialized)
            self.assertEqual(b.is_enabled(), a.is_enabled())
            if a.is_enabled():
                self.assertEqual(b.get_scale(), 3.)
                self.assertEqual(b.get_growth_factor(), 4.)
                self.assertEqual(b.get_backoff_factor(), .5)
                self.assertEqual(b.get_growth_interval(), 2)
                self.assertEqual(b._init_growth_tracker, 0)
                
                self.assertEqual(b._per_optimizer_states["fdsa"],
                                 torch.cuda.amp.grad_scaler._refresh_per_optimizer_state())
                if lazy_init_scale:
                    self.assertEqual(b.scale(torch.tensor([4.0], d=torch.float32, d=d)), 12.0)
    
    def _test_multinomial_empty(self, d, replacement, num_samples):
        probs = torch.ones(0, 3, d=d)
        expected = torch.empty(0, num_samples, d=torch.int64)
        out = torch.multinomial(probs, num_samples=num_samples, replacement=replacement)
        self.assertEqual(out, expected)
    
    def test_multinomial_empty_w_replacement(self, d):
        self._test_multinomial_empty(d, True, 1)
        self._test_multinomial_empty(d, True, 2)
    
    def test_multinomial_empty_wo_replacement(self, d):