chore: import upstream snapshot with attribution

2026-07-13 13:36:55 +08:00
commit c8a779b1bb
1887 changed files with 3245738 additions and 0 deletions
@@ -0,0 +1,153 @@
+#
+# SPDX-FileCopyrightText: Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+---
+name: ScatterElements
+interface: "IPluginV3"
+versions:
+  "2":
+    inputs:
+      - data
+      - indices
+      - updates
+    supported_input_types:
+      - combination1:
+          data: float32
+          indices: int64
+          updates: float32
+      - combination2:
+          data: int32
+          indices: int64
+          updates: int32
+      - combination3:
+          data: int64
+          indices: int64
+          updates: int64
+      - combination4:
+          data: float16
+          indices: int64
+          updates: float16
+      - combination5:
+          data: bfloat16
+          indices: int64
+          updates: bfloat16
+    configs:
+      config1:
+        input_types:
+          data: float32
+          indices: int64
+          updates: float32
+        attribute_options:
+          axis:
+            - -1
+            - 0
+            - 1
+          reduction:
+            - "add"
+            - "mul"
+            - "min"
+            - "max"
+      config2:
+        input_types:
+          data: float16
+          indices: int64
+          updates: float16
+        attribute_options:
+          axis:
+            - -1
+            - 0
+            - 1
+          reduction:
+            - "add"
+            - "mul"
+            - "min"
+            - "max"
+      config3:
+        input_types:
+          data: int32
+          indices: int64
+          updates: int32
+        attribute_options:
+          axis:
+            - -1
+            - 0
+            - 1
+          reduction:
+            - "add"
+            - "mul"
+            - "min"
+            - "max"
+      config4:
+        input_types:
+          data: int64
+          indices: int64
+          updates: int64
+        attribute_options:
+          axis:
+            - -1
+            - 0
+            - 1
+          reduction:
+            - "add"
+            - "mul"
+            - "min"
+            - "max"
+      config5:
+        input_types:
+          data: bfloat16
+          indices: int64
+          updates: bfloat16
+        attribute_options:
+          axis:
+            - -1
+            - 0
+            - 1
+          reduction:
+            - "add"
+            - "mul"
+            - "min"
+            - "max"
+    outputs:
+      - output
+    attributes:
+      - axis
+      - reduction
+    attribute_types:
+      axis: int32
+      reduction: char
+    attribute_length:
+      axis: 1
+      reduction: -1
+    attribute_options:
+      axis:
+        min: "=ninf"
+        max: "=pinf"
+      reduction:
+        - "add"
+        - "mul"
+        - "min"
+        - "max"
+    attributes_required:
+      - reduction
+    golden_io_path: "plugin/ScatterElementsPlugin_PluginGoldenIO.json"
+    abs_tol: 1e-2
+    rel_tol: 1e-2
+    bf16_rtol: 5e-2
+    bf16_atol: 5e-2
+    fp16_rtol: 5e-2
+    fp16_atol: 5e-2
+...