Mark some more ops as pt2_compliant_tag (pytorch#2171)

Summary:
Pull Request resolved: pytorch#2171

Also deleted two skips that were marked as flaky. Those don't appear to
actually be flaky.

Reviewed By: williamwen42

Differential Revision: D51647391

fbshipit-source-id: 4c70558e2e6d4cb7d2c8566e40f5da06050681d4

fbgemm_gpu/src/sparse_ops/sparse_ops_cpu.cpp

@@ -2730,12 +2730,15 @@ TORCH_LIBRARY_FRAGMENT(fbgemm, m) {
   m.def(
       "permute_sparse_data(Tensor permute, Tensor lengths, Tensor values, Tensor? weights=None, SymInt? permuted_lengths_sum=None) -> (Tensor, Tensor, Tensor?)");
   m.def(
-      "permute_2D_sparse_data(Tensor permute, Tensor lengths, Tensor values, Tensor? weights=None, SymInt? permuted_lengths_sum=None) -> (Tensor, Tensor, Tensor?)");
+      "permute_2D_sparse_data(Tensor permute, Tensor lengths, Tensor values, Tensor? weights=None, SymInt? permuted_lengths_sum=None) -> (Tensor, Tensor, Tensor?)",
+      {PT2_COMPLIANT_TAG});
   m.def(
-      "permute_1D_sparse_data(Tensor permute, Tensor lengths, Tensor values, Tensor? weights=None, SymInt? permuted_lengths_sum=None) -> (Tensor, Tensor, Tensor?)");
+      "permute_1D_sparse_data(Tensor permute, Tensor lengths, Tensor values, Tensor? weights=None, SymInt? permuted_lengths_sum=None) -> (Tensor, Tensor, Tensor?)",
+      {PT2_COMPLIANT_TAG});
   m.def("invert_permute(Tensor permute) -> Tensor");
   m.def(
-      "expand_into_jagged_permute(Tensor permute, Tensor input_offset, Tensor output_offset, SymInt output_size) -> Tensor");
+      "expand_into_jagged_permute(Tensor permute, Tensor input_offset, Tensor output_offset, SymInt output_size) -> Tensor",
+      {PT2_COMPLIANT_TAG});
   m.def(
       "block_bucketize_sparse_features(Tensor lengths, Tensor indices, bool bucketize_pos, bool sequence, Tensor block_sizes, SymInt my_size, Tensor? weights=None, Tensor? batch_size_per_feature=None, SymInt max_B= -1, Tensor[]? block_bucketize_pos=None) -> (Tensor, Tensor, Tensor?, Tensor?, Tensor?)");
   m.def(

fbgemm_gpu/test/failures_dict.json

@@ -436,18 +436,8 @@
         "status": "xfail"
       }
     },
-    "fbgemm::permute_1D_sparse_data": {
-      "SparseOpsTest.test_schema__test_permute_indices": {
-        "comment": "flaky",
-        "status": "skip"
-      }
-    },
-    "fbgemm::permute_2D_sparse_data": {
-      "SparseOpsTest.test_schema__test_permute_indices": {
-        "comment": "flaky",
-        "status": "skip"
-      }
-    },
+    "fbgemm::permute_1D_sparse_data": {},
+    "fbgemm::permute_2D_sparse_data": {},
     "fbgemm::permute_sequence_embeddings": {
       "SparseOpsTest.test_aot_dispatch_dynamic__test_permute_embeddings": {
         "comment": "",