From e5d62418af4356269b6b3675be8a97a45e62bb4a Mon Sep 17 00:00:00 2001
From: Yifan Zhao <yifanz16@illinois.edu>
Date: Sun, 4 Apr 2021 00:12:32 -0500
Subject: [PATCH] Fixed depthwise conv in metadata
---
hpvm/projects/torch2hpvm/torch2hpvm/graph_ir.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/hpvm/projects/torch2hpvm/torch2hpvm/graph_ir.py b/hpvm/projects/torch2hpvm/torch2hpvm/graph_ir.py
index 5c248f829a..abf7b60ff3 100644
--- a/hpvm/projects/torch2hpvm/torch2hpvm/graph_ir.py
+++ b/hpvm/projects/torch2hpvm/torch2hpvm/graph_ir.py
@@ -115,7 +115,6 @@ class WeightTensor(TensorNode):
class Conv2DNode(DFGNode):
op_type = "Conv2D"
- hpvm_op_type = "convolution"
def __init__(
self,
@@ -135,6 +134,11 @@ class Conv2DNode(DFGNode):
if list(dilations) != [1, 1]:
raise ValueError("Dilation > 1 is unsupported")
self.group = group
+ if group == 1:
+ self.hpvm_op_type = "convolution"
+ else:
+ # HPVM tensor_runtime distinguishes these two. (sigh)
+ self.hpvm_op_type = "depthwise_convolution"
self.pads = pads[0]
self.sh, self.sw = strides
--
GitLab