From 8068fd1fa5244f4f00f559b8a531b6276403ee2a Mon Sep 17 00:00:00 2001
From: Yifan Zhao <yifanz16@illinois.edu>
Date: Tue, 12 Jan 2021 06:46:33 -0600
Subject: [PATCH] Pass arguments correctly for tensor codegen
---
hpvm/projects/onnx_frontend/main.py | 10 +++++-----
1 file changed, 5 insertions(+), 5 deletions(-)
diff --git a/hpvm/projects/onnx_frontend/main.py b/hpvm/projects/onnx_frontend/main.py
index 07b5bb7249..184877082b 100644
--- a/hpvm/projects/onnx_frontend/main.py
+++ b/hpvm/projects/onnx_frontend/main.py
@@ -41,11 +41,11 @@ def compile(
model = check_version(model, opset)
dfg = DFG(model.graph)
if hpvmc:
- hpvmCodeGen = HpvmCodeGen(dfg, output_dir, input_size, batch_size, prefix)
- hpvmCodeGen.compile()
+ hpvm_code_gen = HpvmCodeGen(dfg, output_dir, input_size, batch_size, prefix)
+ hpvm_code_gen.compile()
else:
- TensorCodeGen = TensorCodeGen(dfg, output_dir, input_size)
- TensorCodeGen.compile()
+ tensor_code_gen = TensorCodeGen(dfg, output_dir, input_size, batch_size, prefix)
+ tensor_code_gen.compile()
dfg.dump_weights(output_dir)
@@ -66,7 +66,7 @@ def parse_args():
"-p",
"--prefix",
type=str,
- help="Prefix in generated code; will be attached before name of weight/input files."
+ help="Prefix in generated code; will be attached before name of weight/input files. "
"Defaults to output_dir.",
)
parser.add_argument(
--
GitLab