Skip to content

Commit 4789d56

Browse files
committed
fix: removed redundant tests
1 parent 5f6156b commit 4789d56

2 files changed

Lines changed: 0 additions & 48 deletions

File tree

tests/function_libs/torch_lib/extra_opinfo.py

Lines changed: 0 additions & 38 deletions
Original file line numberDiff line numberDiff line change
@@ -2483,44 +2483,6 @@ def __init__(self):
24832483
sample_inputs_func=sample_inputs_embedding_bag_padding_idx,
24842484
supports_out=False,
24852485
),
2486-
opinfo_core.OpInfo(
2487-
"ops.aten.embedding_bag.padding_idx_none",
2488-
op=torch.nn.functional.embedding_bag,
2489-
dtypes=common_dtype.floating_types_and_half(),
2490-
sample_inputs_func=lambda op_info, device, dtype, requires_grad: [
2491-
opinfo_core.SampleInput(
2492-
torch.tensor(
2493-
[[1.0, 1.0, 1.0], [2.0, 2.0, 2.0], [3.0, 3.0, 3.0], [4.0, 4.0, 4.0]],
2494-
dtype=dtype,
2495-
device=device,
2496-
),
2497-
args=(
2498-
torch.tensor([0, 1, 2, 3], dtype=torch.int64, device=device),
2499-
torch.tensor([0, 2], dtype=torch.int64, device=device),
2500-
),
2501-
kwargs={"padding_idx": None},
2502-
)
2503-
],
2504-
),
2505-
opinfo_core.OpInfo(
2506-
"ops.aten.embedding_bag.padding_idx_int",
2507-
op=torch.nn.functional.embedding_bag,
2508-
dtypes=common_dtype.floating_types_and_half(),
2509-
sample_inputs_func=lambda op_info, device, dtype, requires_grad: [
2510-
opinfo_core.SampleInput(
2511-
torch.tensor(
2512-
[[1.0, 1.0, 1.0], [2.0, 2.0, 2.0], [3.0, 3.0, 3.0]],
2513-
dtype=dtype,
2514-
device=device,
2515-
),
2516-
args=(
2517-
torch.tensor([0, 1, 2], dtype=torch.int64, device=device),
2518-
torch.tensor([0, 2], dtype=torch.int64, device=device),
2519-
),
2520-
kwargs={"padding_idx": 0},
2521-
)
2522-
],
2523-
),
25242486
opinfo_core.OpInfo(
25252487
"ops.aten.embedding_renorm",
25262488
aten_name="embedding_renorm",

tests/function_libs/torch_lib/ops_test_data.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -932,16 +932,6 @@ def _where_input_wrangler(
932932
dtypes=(torch.float16,),
933933
reason="fixme: results mismatch in torch nightly.",
934934
),
935-
TorchLibOpInfo(
936-
"ops.aten.embedding_bag.padding_idx_none",
937-
core_ops.aten_embedding_bag,
938-
input_wrangler=_embedding_bag_input_wrangler,
939-
),
940-
TorchLibOpInfo(
941-
"ops.aten.embedding_bag.padding_idx_int",
942-
core_ops.aten_embedding_bag,
943-
input_wrangler=_embedding_bag_input_wrangler,
944-
),
945935
TorchLibOpInfo(
946936
"ops.aten.embedding_bag.padding_idx",
947937
core_ops.aten_embedding_bag_padding_idx,

0 commit comments

Comments
 (0)