Skip to content

Commit

Permalink
Skip test_pack_segments until the issue is fixed (pytorch#2032)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: pytorch#2032

`test_pack_segments` fails on cpu CI. Skip the test temporarily to unblock nightly releases while the issue is being investigated.

https://github.com/pytorch/FBGEMM/actions/runs/6254900720

Reviewed By: q10

Differential Revision: D49524460

fbshipit-source-id: 1c8f976dbed280c989ad0cf6fd14e852473cd686
  • Loading branch information
spcyppt authored and facebook-github-bot committed Sep 22, 2023
1 parent 8f02dd3 commit 91facab
Showing 1 changed file with 8 additions and 2 deletions.
10 changes: 8 additions & 2 deletions fbgemm_gpu/test/sparse_ops_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,12 +26,17 @@
from fbgemm_gpu import open_source # noqa: F401

# pyre-ignore[21]
from test_utils import gpu_available, gpu_unavailable, skipIfRocm
from test_utils import gpu_available, gpu_unavailable, running_on_github, skipIfRocm
except Exception:
torch.ops.load_library("//deeplearning/fbgemm/fbgemm_gpu:sparse_ops")
torch.ops.load_library("//deeplearning/fbgemm/fbgemm_gpu:sparse_ops_cpu")
torch.ops.load_library("//deeplearning/fbgemm/fbgemm_gpu/codegen:index_select_ops")
from fbgemm_gpu.test.test_utils import gpu_available, gpu_unavailable, skipIfRocm
from fbgemm_gpu.test.test_utils import (
gpu_available,
gpu_unavailable,
running_on_github,
skipIfRocm,
)


def unbucketize_indices_value(
Expand Down Expand Up @@ -1733,6 +1738,7 @@ def _pack_segments_ref(
),
torch_compile=st.booleans(),
)
@unittest.skipIf(*running_on_github)
@settings(deadline=None)
def test_pack_segments(
self,
Expand Down

0 comments on commit 91facab

Please sign in to comment.