From 91c97a52598160affd3dc2245ce608656a16ad52 Mon Sep 17 00:00:00 2001
From: Yifan Zhao <yifanz16@illinois.edu>
Date: Thu, 18 Mar 2021 22:20:49 -0500
Subject: [PATCH] Fixed operator names when exporting to hpvm config
---
predtuner/pipedbin.py | 8 +++++++-
1 file changed, 7 insertions(+), 1 deletion(-)
diff --git a/predtuner/pipedbin.py b/predtuner/pipedbin.py
index b8cf173..18237d1 100644
--- a/predtuner/pipedbin.py
+++ b/predtuner/pipedbin.py
@@ -253,7 +253,13 @@ class HPVMConfigBuilder:
["maxpool"],
]
- op_to_op = {"convolution": "conv", "maxpool": "pool_max", "linear": "mul"}
+ op_to_op = {
+ "convolution": "conv",
+ "maxpool": "pool_max",
+ "linear": "mul",
+ "avgpool": "pool_mean",
+ "depthwise_convolution": "group_conv"
+ }
knob_name_to_range = {
"fp32": range(11, 12),
--
GitLab