Skip to content
Permalink

Comparing changes

This is a direct comparison between two commits made in this repository or its related repositories. View the default comparison for this range or learn more about diff comparisons.

Open a pull request

Create a new pull request by comparing changes across two branches. If you need to, you can also . Learn more about diff comparisons here.
base repository: nod-ai/SHARK-TestSuite
Failed to load repositories. Confirm that selected base ref is valid, then try again.
Loading
base: dd147896e97229eff2065c3fd90f075689f70faa
Choose a base ref
..
head repository: nod-ai/SHARK-TestSuite
Failed to load repositories. Confirm that selected head ref is valid, then try again.
Loading
compare: e0f4df0be5d1822045c28f80f76677d96a1b61bd
Choose a head ref
Showing with 6 additions and 4 deletions.
  1. +6 −4 alt_e2eshark/onnx_tests/models/nlp.py
10 changes: 6 additions & 4 deletions alt_e2eshark/onnx_tests/models/nlp.py
View file
Original file line number Diff line number Diff line change
@@ -124,7 +124,8 @@ def construct_inputs(self):
# running test mygpt4...
# Unzipping - /proj/gdba/shark/chi/src/SHARK-TestSuite/alt_e2eshark/tmp/mygpt4/model.onnx.zip...
# Unzipping succeded. Look for extracted contents in /proj/gdba/shark/chi/src/SHARK-TestSuite/alt_e2eshark/test-run/mygpt4
# {'Shape': 63, 'Cast': 141, 'Slice': 136, 'Squeeze': 25, 'Range': 25, 'Concat': 65, 'Reshape': 161, 'Unsqueeze': 38, 'Sub': 85, 'Mul': 169, 'Add': 159, 'Gather': 3, 'MatMul': 72, 'Split': 12, 'Transpose': 48, 'GreaterOrEqual': 12, 'Softmax': 12, 'GlobalAveragePool': 48, 'Sqrt': 24, 'Reciprocal': 24, 'Erf': 12}# Running stage 'import_model'...
# {'Shape': 63, 'Cast': 141, 'Slice': 136, 'Squeeze': 25, 'Range': 25, 'Concat': 65, 'Reshape': 161, 'Unsqueeze': 38, 'Sub': 85, 'Mul': 169, 'Add': 159, 'Gather': 3, 'MatMul': 72, 'Split': 12, 'Transpose': 48, 'GreaterOrEqual': 12, 'Softmax': 12, 'GlobalAveragePool': 48, 'Sqrt': 24, 'Reciprocal': 24, 'Erf': 12}#
# Running stage 'import_model'...
register_test(t_model_constructor(1, ""), "mygpt4")
# Run to the last 2/7/12/17 onnx op with:
# run with python ./run.py --mode=cl-onnx-iree -v -t mygpt4_trunc_
@@ -133,8 +134,9 @@ def construct_inputs(self):

# Run to 5/55/105/155 onnx.Add ops with:
# run with python ./run.py --mode=cl-onnx-iree -v -t mygpt4_trunc_add_
# for n in range(5, 160, 50):
# register_test(t_model_constructor(n,"Add"), f"mygpt4_trunc_add_{n}")
for n in range(5, 160, 50):
register_test(t_model_constructor(n,"Add"), f"mygpt4_trunc_add_{n}")

# python ./run.py --mode=cl-onnx-iree -v -t mygpt4_trunc_shape_1
register_test(t_model_constructor(1, "Shape"), "mygpt4_trunc_shape_1")
register_test(t_model_constructor(1, "Shape"), "mygpt4_trunc_shape_1")
register_test(t_model_constructor(161, "Reshape"), "mygpt4_trunc_reshape_161")