diff --git a/hpvm/projects/onnx_frontend/frontend/codegen_hpvm.py b/hpvm/projects/onnx_frontend/frontend/codegen_hpvm.py index 1a130fc30bec685ee7c903beb35494e939ed25d3..fbdf0aea083ed34ae0f89b51426747a97272380b 100644 --- a/hpvm/projects/onnx_frontend/frontend/codegen_hpvm.py +++ b/hpvm/projects/onnx_frontend/frontend/codegen_hpvm.py @@ -7,7 +7,7 @@ import jinja2 from .graph_builder import DFG from .graph_ir import DFGNode, TensorNode, WeightTensor -TEMPLATE_FILE = "template_hpvm.cpp" +TEMPLATE_FILE = "template_hpvm.cpp.in" loader = jinja2.FileSystemLoader(searchpath=Path(__file__).parent) template_env = jinja2.Environment(loader=loader, trim_blocks=True) template = template_env.get_template(TEMPLATE_FILE) diff --git a/hpvm/projects/onnx_frontend/frontend/codegen_tensor.py b/hpvm/projects/onnx_frontend/frontend/codegen_tensor.py index c4660b24bad43c3ed8928248a17211e1ee291da9..44058b0ac454cca150c19fc568fde4b9d726d199 100644 --- a/hpvm/projects/onnx_frontend/frontend/codegen_tensor.py +++ b/hpvm/projects/onnx_frontend/frontend/codegen_tensor.py @@ -6,7 +6,7 @@ import jinja2 from .codegen_hpvm import CodeGen from .graph_ir import DFGNode, TensorNode -TEMPLATE_FILE = "template_tensor.cpp" +TEMPLATE_FILE = "template_tensor.cpp.in" loader = jinja2.FileSystemLoader(searchpath=Path(__file__).parent) template_env = jinja2.Environment(loader=loader, trim_blocks=True) template = template_env.get_template(TEMPLATE_FILE) diff --git a/hpvm/projects/onnx_frontend/frontend/template_hpvm.cpp b/hpvm/projects/onnx_frontend/frontend/template_hpvm.cpp.in similarity index 100% rename from hpvm/projects/onnx_frontend/frontend/template_hpvm.cpp rename to hpvm/projects/onnx_frontend/frontend/template_hpvm.cpp.in diff --git a/hpvm/projects/onnx_frontend/frontend/template_tensor.cpp b/hpvm/projects/onnx_frontend/frontend/template_tensor.cpp.in similarity index 100% rename from hpvm/projects/onnx_frontend/frontend/template_tensor.cpp rename to hpvm/projects/onnx_frontend/frontend/template_tensor.cpp.in