From 0b239e5f0ad3a940a959b715fc60ec8597c2873f Mon Sep 17 00:00:00 2001
From: Yifan Zhao <yifanz16@illinois.edu>
Date: Tue, 2 Feb 2021 07:05:50 -0600
Subject: [PATCH] Fixed setup.py requirement
---
hpvm/projects/torch2hpvm/setup.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/hpvm/projects/torch2hpvm/setup.py b/hpvm/projects/torch2hpvm/setup.py
index ae103a2cdf..f0cd851e58 100644
--- a/hpvm/projects/torch2hpvm/setup.py
+++ b/hpvm/projects/torch2hpvm/setup.py
@@ -7,6 +7,8 @@ setup(
author="Yuanjing Shi, Yifan Zhao",
author_email="ys26@illinois.edu, yifanz16@illinois.edu",
packages=["torch2hpvm"],
- install_requires=["jinja2>=2.11", "networkx>=2.5", "onnx>=1.8.0", "torch"],
+ install_requires=[
+ "jinja2>=2.11", "networkx>=2.5", "onnx>=1.8.0", "torch", "onnx-simplifier>=0.2.27"
+ ],
entry_points={"console_scripts": ["torch2hpvm=torch2hpvm:main"]},
)
--
GitLab