File tree Expand file tree Collapse file tree 1 file changed +7
-5
lines changed
Expand file tree Collapse file tree 1 file changed +7
-5
lines changed Original file line number Diff line number Diff line change @@ -36,10 +36,10 @@ def get_hardward_name(args):
3636
3737def get_compile_framework_version (args ):
3838 if args .compiler == "cinn" :
39- compile_framework_version = paddle .__version__
40- else :
41- compile_framework_version = "unknown "
42- return compile_framework_version
39+ return paddle .__version__
40+ if args . compiler == "nope" :
41+ return "nope-baseline "
42+ return "unknown"
4343
4444
4545def load_class_from_file (file_path : str , class_name : str ):
@@ -92,6 +92,8 @@ def get_input_spec(args):
9292
9393
9494def get_compiled_model (args , model ):
95+ if args .compiler == "nope" :
96+ return model
9597 input_spec = get_input_spec (args )
9698 build_strategy = paddle .static .BuildStrategy ()
9799 compiled_model = paddle .jit .to_static (
@@ -385,7 +387,7 @@ def test_multi_models(args):
385387
386388def main (args ):
387389 assert os .path .isdir (args .model_path )
388- assert args .compiler == "cinn"
390+ assert args .compiler in { "cinn" , "nope" }
389391
390392 initalize_seed = 123
391393 set_seed (random_seed = initalize_seed )
You can’t perform that action at this time.
0 commit comments