Commita54de21

authored

Better support for ir_version (#82)

* fixes for ir_version* fix ut* fix ut

1 parent492b6d4 commita54de21Copy full SHA for a54de21

File tree

+11

-2

lines changed

+11

-2

lines changed

Lines changed: 3 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -242,7 +242,7 @@ def run_node(cls, node, inputs, device=None, outputs_info=None, **kwargs):`
`242`	`242`
`243`	`243`	`# The following tests are too slow with the reference implementation (Conv).`
`244`	`244`	`backend_test.exclude(`
`245`		`-"(FLOAT8\|BFLOAT16\|_opt_\|_3d_\|_momentum_\|_4d_"`
	`245`	`+"(FLOAT8\|BFLOAT16\|INT4\|_opt_\|_3d_\|_momentum_\|_4d_\|int4"`
`246`	`246`	`"\|test_adagrad"`
`247`	`247`	`"\|test_adam"`
`248`	`248`	`"\|test_ai_onnx_ml_"`
`@@ -270,6 +270,8 @@ def run_node(cls, node, inputs, device=None, outputs_info=None, **kwargs):`
`270`	`270`	`"\|test_squeezenet"`
`271`	`271`	`"\|test_vgg19"`
`272`	`272`	`"\|test_zfnet512"`
	`273`	`+"\|test_range_float_type_positive_delta_expanded"`
	`274`	`+"\|test_range_int32_type_negative_delta_expanded"`
`273`	`275`	`")"`
`274`	`276`	`)`
`275`	`277`

Lines changed: 3 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -149,7 +149,7 @@ def run_node(cls, node, inputs, device=None, outputs_info=None, **kwargs):`
`149`	`149`	`"\|test_scan_sum)"`
`150`	`150`	`)`
`151`	`151`
`152`		`-ifonnx_opset_version()<21:`
	`152`	`+ifonnx_opset_version()<200:`
`153`	`153`	`# The following tests are using types not supported by NumPy.`
`154`	`154`	`# They could be if method to_array is extended to support custom`
`155`	`155`	`# types the same as the reference implementation does`
`@@ -164,8 +164,10 @@ def run_node(cls, node, inputs, device=None, outputs_info=None, **kwargs):`
`164`	`164`	`"\|test_cast_no_saturate_"`
`165`	`165`	`"\|_to_FLOAT8"`
`166`	`166`	`"\|_FLOAT8"`
	`167`	`+"\|INT4"`
`167`	`168`	`"\|test_quantizelinear_e4m3fn"`
`168`	`169`	`"\|test_quantizelinear_e5m2"`
	`170`	`+"\|test_scatter_with"`
`169`	`171`	`")"`
`170`	`172`	`)`
`171`	`173`

Lines changed: 5 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -156,6 +156,7 @@ def __init__(`
`156`	`156`	`optimization_options:Optional[OptimizationOptions]=None,`
`157`	`157`	`args:Optional[List[Any]]=None,`
`158`	`158`	`verbose:int=0,`
	`159`	`+ir_version:Optional[int]=None,`
`159`	`160`	`):`
`160`	`161`	`self.optimization_options=optimization_optionsorOptimizationOptions()`
`161`	`162`	`self.as_function=as_function`
`@@ -170,6 +171,7 @@ def __init__(`
`170`	`171`	`ifisinstance(target_opset_or_existing_proto,int)`
`171`	`172`	`elsetarget_opset_or_existing_proto`
`172`	`173`	`)`
	`174`	`+self.ir_version=ir_version`
`173`	`175`	`self.nodes= []`
`174`	`176`	`self.initializers_dict= {}`
`175`	`177`	`self.inputs= []`
`@@ -186,6 +188,7 @@ def __init__(`
`186`	`188`	`),"input_names must be empty if the input is an existing model."`
`187`	`189`	`proto=target_opset_or_existing_proto`
`188`	`190`	`self.opsets= {d.domain:d.versionfordinproto.opset_import}`
	`191`	`+self.ir_version=ir_versionortarget_opset_or_existing_proto.ir_version`
`189`	`192`	`self.nodes=list(proto.graph.node)`
`190`	`193`	`self.initializers_dict= {i.name:iforiinproto.graph.initializer}`
`191`	`194`	`self.initializers_dict.update(`
`@@ -674,6 +677,8 @@ def to_onnx(`
`674`	`677`	`ifself.verbose:`
`675`	`678`	`print("[GraphBuilder] onh.make_model")`
`676`	`679`	`model=oh.make_model(graph,opset_imports=opsets)`
	`680`	`+ifself.ir_version:`
	`681`	`+model.ir_version=self.ir_version`
`677`	`682`	`returnmodel`
`678`	`683`
`679`	`684`	`def_check_order_node(self,ind:int,node:NodeProto,existing:Set[str]):`

Comments

(0)