From 711cc55ca70781d3340f6c7e663825f8f68c3205 Mon Sep 17 00:00:00 2001 From: Manfei <41607353+ManfeiBai@users.noreply.github.com> Date: Sun, 15 Sep 2024 01:37:25 -0700 Subject: [PATCH] add `masked.median` --- experimental/torch_xla2/test/test_ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/experimental/torch_xla2/test/test_ops.py b/experimental/torch_xla2/test/test_ops.py index 2b88ae324fa..ac2e0fdb94b 100644 --- a/experimental/torch_xla2/test/test_ops.py +++ b/experimental/torch_xla2/test/test_ops.py @@ -114,7 +114,7 @@ "masked.logaddexp", "masked.logsumexp", "masked.mean", - "masked.median", + "masked.median", # RuntimeError: Object of type 'XLATensor2' is not an instance of 'sequence' "masked_scatter", "masked_select", "max_pool2d_with_indices_backward",