Re-enable jagged_tensor_ops tests on arm (pytorch#2210)

Summary:
As titled.
Pull Request resolved: pytorch#2210

Reviewed By: q10

Differential Revision: D52107179

fbshipit-source-id: c5d2e9de6edd8a072d03d731474ac574270ba78d

fbgemm_gpu/test/jagged_tensor_ops_test.py

@@ -28,7 +28,6 @@
         gpu_available,
         gpu_unavailable,
         gradcheck,
-        on_arm_platform,
         optests,
         symint_vector_unsupported,
         TEST_WITH_ROCM,
@@ -41,7 +40,6 @@
         gpu_available,
         gpu_unavailable,
         gradcheck,
-        on_arm_platform,
         optests,
         symint_vector_unsupported,
         TEST_WITH_ROCM,
@@ -2365,7 +2363,6 @@ def test_jagged_softmax(
         if gpu_available
         else st.just("cpu"),
     )
-    @unittest.skipIf(*on_arm_platform)
     @settings(verbosity=Verbosity.verbose, max_examples=20, deadline=None)
     def test_jagged_jagged_bmm(
         self,
@@ -2431,7 +2428,6 @@ def test_jagged_jagged_bmm(
         if gpu_available
         else st.just("cpu"),
     )
-    @unittest.skipIf(*on_arm_platform)
     @settings(verbosity=Verbosity.verbose, max_examples=2, deadline=None)
     def test_jagged_dense_bmm(
         self,
@@ -2494,7 +2490,6 @@ def test_jagged_dense_bmm(
         dtype=st.sampled_from([torch.float, torch.double]),
         device_type=st.just("cpu"),
     )
-    @unittest.skipIf(*on_arm_platform)
     @settings(verbosity=Verbosity.verbose, max_examples=20, deadline=None)
     def test_jagged_dense_bmm_dynamic_shape(
         self,