Add Op(embedding_bag_padding_idx) | torchlib(feat)

[xiaowuhu]

1. It works for common test case, say, when offsets value is common like [0,2,3], but something like [0,2,2,4] will be failed. Guess it is due to op.While ‘s bug.
2. Only care the shape of the last 3 outputs.
3. Test case not contain 2D indices, wait to see if it is necessary.
4. Need to disable loop/scan constrain checking

Fixes #1021

[codecov]

Codecov Report

Merging #1022 (affdc13) into main (0c1995d) will increase coverage by 0.13%.
The diff coverage is 100.00%.

@@            Coverage Diff             @@
##             main    #1022      +/-   ##
==========================================
+ Coverage   77.50%   77.63%   +0.13%     
==========================================
  Files         114      114              
  Lines       14335    14423      +88     
  Branches     1525     1543      +18     
==========================================
+ Hits        11110    11198      +88     
  Misses       2859     2859              
  Partials      366      366              

Files Changed	Coverage Δ
...bs/tools/torch_lib/deduce_type_constraints_test.py	86.66% <ø> (ø)
...ipt/tests/function_libs/torch_lib/ops_test_data.py	96.00% <ø> (ø)
onnxscript/function_libs/torch_lib/ops/core.py	78.94% <100.00%> (+0.49%)	⬆️
...ript/tests/function_libs/torch_lib/extra_opinfo.py	98.21% <100.00%> (+0.17%)	⬆️

[github-actions]

Test Results

         18 files  ±  0         18 suites  ±0   1h 19m 16s ⏱️ - 5m 7s
  10 572 tests +  6    7 853 ✔️ +  5      2 715 💤 +  1  3 ❌ ±0  1 🔥 ±0 
156 149 runs  +60  36 626 ✔️ +50  119 513 💤 +10  9 ❌ ±0  1 🔥 ±0 

For more details on these failures and errors, see this check.

Results for commit affdc13. ± Comparison against base commit 0c1995d.

This pull request removes 129 and adds 135 tests. Note that renamed tests count towards both.

onnxscript.tests.function_libs.torch_lib.ops_test.TestFunctionValidity ‑ test_function_has_op_schema_137_aten_embedding
onnxscript.tests.function_libs.torch_lib.ops_test.TestFunctionValidity ‑ test_function_has_op_schema_138_aten_hardtanh
onnxscript.tests.function_libs.torch_lib.ops_test.TestFunctionValidity ‑ test_function_has_op_schema_139_aten_leaky_relu
onnxscript.tests.function_libs.torch_lib.ops_test.TestFunctionValidity ‑ test_function_has_op_schema_140_aten_log_sigmoid
onnxscript.tests.function_libs.torch_lib.ops_test.TestFunctionValidity ‑ test_function_has_op_schema_141_aten_nll_loss_weight
onnxscript.tests.function_libs.torch_lib.ops_test.TestFunctionValidity ‑ test_function_has_op_schema_142_aten_nll_loss
onnxscript.tests.function_libs.torch_lib.ops_test.TestFunctionValidity ‑ test_function_has_op_schema_143_aten_reflection_pad2d
onnxscript.tests.function_libs.torch_lib.ops_test.TestFunctionValidity ‑ test_function_has_op_schema_144_aten_relu
onnxscript.tests.function_libs.torch_lib.ops_test.TestFunctionValidity ‑ test_function_has_op_schema_145_aten_relu6
onnxscript.tests.function_libs.torch_lib.ops_test.TestFunctionValidity ‑ test_function_has_op_schema_146_aten_replication_pad2d
…

onnxscript.function_libs.tools.torch_lib.deduce_type_constraints_test.TestDeduceTypeConstraints ‑ test_deduce_type_constraints_does_not_crash_for_onnx_function__aten_embedding_bag_1d_padding_idx_onnx
onnxscript.tests.function_libs.torch_lib.ops_test.TestFunctionValidity ‑ test_function_has_op_schema_137_aten_embedding_bag_padding_idx
onnxscript.tests.function_libs.torch_lib.ops_test.TestFunctionValidity ‑ test_function_has_op_schema_138_aten_embedding
onnxscript.tests.function_libs.torch_lib.ops_test.TestFunctionValidity ‑ test_function_has_op_schema_139_aten_hardtanh
onnxscript.tests.function_libs.torch_lib.ops_test.TestFunctionValidity ‑ test_function_has_op_schema_140_aten_leaky_relu
onnxscript.tests.function_libs.torch_lib.ops_test.TestFunctionValidity ‑ test_function_has_op_schema_141_aten_log_sigmoid
onnxscript.tests.function_libs.torch_lib.ops_test.TestFunctionValidity ‑ test_function_has_op_schema_142_aten_nll_loss_weight
onnxscript.tests.function_libs.torch_lib.ops_test.TestFunctionValidity ‑ test_function_has_op_schema_143_aten_nll_loss
onnxscript.tests.function_libs.torch_lib.ops_test.TestFunctionValidity ‑ test_function_has_op_schema_144_aten_reflection_pad2d
onnxscript.tests.function_libs.torch_lib.ops_test.TestFunctionValidity ‑ test_function_has_op_schema_145_aten_relu
…

♻️ This comment has been updated with latest results.

[justinchuby]

Awesome! 🚀