add copyright, add diffs for minicpm

CuriousPanCake · Oct 17, 2024 · 1824a80 · 1824a80
1 parent e60122d
commit 1824a80
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/tests/model_hub_tests/transformation_tests/generate_ref_diffs.py b/tests/model_hub_tests/transformation_tests/generate_ref_diffs.py
@@ -1,3 +1,6 @@
+# Copyright (C) 2018-2024 Intel Corporation
+# SPDX-License-Identifier: Apache-2.0
+
 '''
 Use this script if you need to regenerate reference diffs for each model
 to test SDPAToPA transformation.

diff --git a/tests/model_hub_tests/transformation_tests/sdpa2pa_ref_diff.py b/tests/model_hub_tests/transformation_tests/sdpa2pa_ref_diff.py
@@ -1,3 +1,6 @@
+# Copyright (C) 2018-2024 Intel Corporation
+# SPDX-License-Identifier: Apache-2.0
+
 nodes_to_compare = ("ScaledDotProductAttention", "PagedAttentionExtension", "Parameter", "ReadValue", "Assign")
 
 ref_diff_map = {
@@ -225,6 +228,13 @@
 		"ReadValue" : -8,
 		"Assign" : -8,
 	},
+    "katuni4ka/tiny-random-minicpm" : {
+        "ReadValue" : -8,
+        "ScaledDotProductAttention" : -4,
+        "Assign" : -8,
+        "PagedAttentionExtension" : 4,
+        "Parameter" : 11,
+    },
 	"katuni4ka/tiny-random-falcon-40b" : {
 		"PagedAttentionExtension" : 2,
 		"ScaledDotProductAttention" : -2,
@@ -522,6 +532,13 @@
 		"Parameter" : 14,
 		"Assign" : -8,
 	},
+    "katuni4ka/tiny-random-minicpm" : {
+        "ScaledDotProductAttention" : -4,
+        "Parameter" : 14,
+        "PagedAttentionExtension" : 4,
+        "ReadValue" : -8,
+        "Assign" : -8,
+    },
 	"katuni4ka/tiny-random-falcon-40b" : {
 		"ScaledDotProductAttention" : -2,
 		"ReadValue" : -4,