Skip to content

Commit

Permalink
[dtensor] update op db tests (pytorch#115722)
Browse files Browse the repository at this point in the history
This PR updates the op db tests xfails, we should see whether we can
enable this again in CI

Pull Request resolved: pytorch#115722
Approved by: https://github.com/XilunWu
  • Loading branch information
wanchaol authored and pytorchmergebot committed Dec 14, 2023
1 parent ef6a0fa commit 49d826b
Showing 1 changed file with 10 additions and 1 deletion.
11 changes: 10 additions & 1 deletion test/distributed/_tensor/test_dtensor_ops.py
Original file line number Diff line number Diff line change
Expand Up @@ -131,7 +131,6 @@ def wrapped(fn):
xfail("constant_pad_nd"),
xfail("corrcoef"),
xfail("count_nonzero"),
xfail("cov"),
xfail("cross"),
xfail("cummax"),
xfail("cummin"),
Expand All @@ -147,9 +146,11 @@ def wrapped(fn):
xfail("dot"),
xfail("einsum"),
xfail("empty"),
xfail("empty_strided"),
xfail("empty_like"),
xfail("empty_permuted"),
xfail("exponential"),
xfail("equal"),
xfail("eye"),
xfail("fft.fft2"),
xfail("fft.fft"),
Expand Down Expand Up @@ -241,10 +242,12 @@ def wrapped(fn):
xfail("linalg.vecdot"),
xfail("linalg.vector_norm"),
xfail("linspace"),
xfail("linspace", "tensor_overload"),
xfail("log_normal"),
xfail("logcumsumexp"),
xfail("logdet"),
xfail("logspace"),
xfail("logspace", "tensor_overload"),
xfail("logsumexp"),
xfail("lu"),
xfail("lu_solve"),
Expand Down Expand Up @@ -279,6 +282,7 @@ def wrapped(fn):
xfail("nanquantile"),
xfail("nansum"),
xfail("native_batch_norm"),
xfail("native_dropout_backward"),
xfail("narrow_copy"),
xfail("ne"),
xfail("new_empty"),
Expand All @@ -302,6 +306,7 @@ def wrapped(fn):
xfail("nn.functional.celu"),
xfail("nn.functional.conv1d"),
xfail("nn.functional.conv2d"),
xfail("nn.functional.conv3d"),
xfail("nn.functional.conv_transpose1d"),
xfail("nn.functional.conv_transpose2d"),
xfail("nn.functional.conv_transpose3d"),
Expand Down Expand Up @@ -329,6 +334,7 @@ def wrapped(fn):
xfail("nn.functional.interpolate", "bilinear"),
xfail("nn.functional.interpolate", "linear"),
xfail("nn.functional.interpolate", "nearest"),
xfail("nn.functional.interpolate", "nearest-exact"),
xfail("nn.functional.interpolate", "trilinear"),
xfail("nn.functional.leaky_relu"),
xfail("nn.functional.linear"),
Expand All @@ -347,13 +353,15 @@ def wrapped(fn):
xfail("nn.functional.mish"),
xfail("nn.functional.mse_loss"),
xfail("nn.functional.multi_margin_loss"),
xfail("nn.functional.multi_head_attention_forward"),
xfail("nn.functional.multilabel_margin_loss"),
xfail("nn.functional.multilabel_soft_margin_loss"),
xfail("nn.functional.nll_loss"),
xfail("nn.functional.normalize"),
xfail("nn.functional.pad", "constant"),
xfail("nn.functional.pad", "reflect"),
xfail("nn.functional.pad", "replicate"),
xfail("nn.functional.pad", "replicate_negative"),
xfail("nn.functional.pairwise_distance"),
xfail("nn.functional.pdist"),
xfail("nn.functional.pixel_shuffle"),
Expand Down Expand Up @@ -482,6 +490,7 @@ def wrapped(fn):
xfail("unique_consecutive"),
xfail("unique"),
xfail("unsafe_split"),
xfail("unsafe_chunk"),
xfail("var_mean"),
xfail("var_mean", "unbiased"),
xfail("vdot"),
Expand Down

0 comments on commit 49d826b

Please sign in to comment.