Enable CPP Extension Open Registration tests on Arm (pytorch#144774)

Enables most tests under CPP Extension Open Registration as they pass on Arm now.

Pull Request resolved: pytorch#144774
Approved by: https://github.com/aditew01, https://github.com/fadara01, https://github.com/malfet

Author: murste01

.ci/pytorch/test.sh

@@ -1526,7 +1526,7 @@ test_executorch() {
 test_linux_aarch64() {
   python test/run_test.py --include test_modules test_mkldnn test_mkldnn_fusion test_openmp test_torch test_dynamic_shapes \
         test_transformers test_multiprocessing test_numpy_interop test_autograd test_binary_ufuncs test_complex test_spectral_ops \
-        test_foreach test_reductions test_unary_ufuncs test_tensor_creation_ops test_ops \
+        test_foreach test_reductions test_unary_ufuncs test_tensor_creation_ops test_ops test_cpp_extensions_open_device_registration \
         --shard "$SHARD_NUMBER" "$NUM_TEST_SHARDS" --verbose
 
   # Dynamo tests

test/test_cpp_extensions_open_device_registration.py

@@ -16,7 +16,6 @@
 import torch.utils.cpp_extension
 from torch.serialization import safe_globals
 from torch.testing._internal.common_utils import (
-    IS_ARM64,
     skipIfTorchDynamo,
     TemporaryFileName,
     TEST_CUDA,
@@ -36,7 +35,6 @@ class _OpenRegMod:
     return _OpenRegMod()
 
 
-@unittest.skipIf(IS_ARM64, "Does not work on arm")
 @unittest.skipIf(TEST_XPU, "XPU does not support cppextension currently")
 @torch.testing._internal.common_utils.markDynamoStrictTest
 class TestCppExtensionOpenRgistration(common.TestCase):