skip test_symeig if MAGMA not detected (pytorch#54526)

Summary:
Add proper way to skip test_symeig. In case MAGMA is not detected, skip the test_symeig properly.
Added skipCUDAIfNoMagma decorator.

Pull Request resolved: pytorch#54526

Reviewed By: malfet

Differential Revision: D27293640

Pulled By: heitorschueroff

fbshipit-source-id: 245f86540af0e37c8795e80dc003e1ca4c08cd5b

Author: arindamroy-eng

test/test_vmap.py

@@ -5,7 +5,8 @@
 import functools
 import itertools
 import warnings
-from torch.testing._internal.common_device_type import instantiate_device_type_tests
+from torch.testing._internal.common_device_type import instantiate_device_type_tests, \
+    skipCUDAIfNoMagma
 import types
 
 
@@ -2409,6 +2410,7 @@ def test_trace(self, device):
         x = torch.randn(2, 3, device=device, requires_grad=True)
         self._batched_grad_test(Tensor.trace, (x,))
 
+    @skipCUDAIfNoMagma
     @allowVmapFallbackUsage
     def test_symeig(self, device):
         def op(x):