Skip to content

Commit

Permalink
#0: disable layernorm pytest due to WH error
Browse files Browse the repository at this point in the history
  • Loading branch information
TT-billteng committed Dec 1, 2023
1 parent 6fb074e commit cb5bc55
Showing 1 changed file with 1 addition and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,7 @@ def test_softmax(shape, device):


@pytest.mark.parametrize("shape", [[1, 1, 32, 32], [1, 1, 32, 128]])
@skip_for_wormhole_b0("Watcher error, see issue#4107")
def test_layernorm(shape, device):
torch.manual_seed(1234)
x = torch.randn(shape).bfloat16().float()
Expand Down

0 comments on commit cb5bc55

Please sign in to comment.