fixes

Author: Aravind-11

onnxscript/function_libs/torch_lib/ops/core.py

@@ -3114,6 +3114,8 @@ def aten_embedding_bag_padding_idx(
         per_sample_weights = op.CastLike(per_sample_weights, weight)
 
     if padding_idx is not None:
+        if padding_idx < 0:
+            padding_idx = weight.shape[0] + padding_idx
         # Call the existing function for handling padding_idx
         result, offset2bag, bag_size, max_indices = _aten_embedding_bag_1d_padding_idx_onnx(
             weight,

tests/function_libs/torch_lib/ops_test_data.py

@@ -1066,7 +1066,7 @@ def _where_input_wrangler(
     ),
     TorchLibOpInfo(
         "ops.aten.embedding_bag.padding_idx_int",
-        core_ops.aten_embedding_bag,
+        core_ops.aten_embedding_bag_padding_idx,
         input_wrangler=_embedding_bag_input_wrangler,
     ),
     TorchLibOpInfo(