Skip to content

Commit 2e34f17

Browse files
committed
Add nope in paddle
1 parent 18154b3 commit 2e34f17

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

graph_net/paddle/test_compiler.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -36,10 +36,10 @@ def get_hardward_name(args):
3636

3737
def 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

4545
def load_class_from_file(file_path: str, class_name: str):
@@ -92,6 +92,8 @@ def get_input_spec(args):
9292

9393

9494
def 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

386388
def 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)

0 commit comments

Comments
 (0)