@@ -33,19 +33,22 @@ class TestSpecDBOperationsMPS(BaseSpecDBTest):
3333 # not implemented
3434 "_pdist_forward.default" ,
3535 # impl: clamp tensor number of dims must not be greater than that of input tensor
36+ # https://github.com/pytorch/pytorch/issues/160734
3637 "clamp.Tensor" ,
3738 # crash: https://github.com/pytorch/pytorch/issues/154881
3839 "cumsum.default" ,
3940 # sparse_grad not supported in MPS yet
4041 "gather.default" ,
4142 # Dimension specified as -1 but tensor has no dimensions
43+ # https://github.com/pytorch/pytorch/issues/160737
4244 "index_select.default" ,
4345 # crash: https://github.com/pytorch/pytorch/issues/154882
4446 "max_pool2d_with_indices.default" ,
4547 # On-going issue on MPSGraph topk when ndims() - axis > 4, see issue #154890
4648 # https://github.com/pytorch/pytorch/issues/154890
4749 "topk.default" ,
4850 # var_mps: reduction dim must be in the range of input shape
51+ # https://github.com/pytorch/pytorch/issues/160738
4952 "var.correction" ,
5053 "var.dim" ,
5154 ]
@@ -59,6 +62,7 @@ def test_all_ops_mps(self):
5962 def test_all_ops_mps_half (self ):
6063 skip_ops = self .SKIP_OPS .copy ()
6164 # ConvTranspose 3D with BF16 or FP16 types is not supported on MPS
65+ # https://github.com/pytorch/pytorch/issues/160739
6266 skip_ops += ["convolution.default" ]
6367
6468 config = TensorConfig (
@@ -70,6 +74,7 @@ def test_all_ops_mps_transposed(self):
7074 skip_ops = self .SKIP_OPS .copy ()
7175 # argmax.default ['ArgType.Tensor torch.float32 (8, 8)', 'ArgType.DimOpt None', 'ArgType.Bool True']
7276 # Exception occurred: view size is not compatible with input tensor's size and stride (at least one dimension spans across two contiguous subspaces). Use .reshape(...) instead.
77+ # https://github.com/pytorch/pytorch/issues/160740
7378 skip_ops += ["argmax.default" , "argmin.default" ]
7479
7580 config = TensorConfig (
0 commit comments