diff --git a/predtuner/pipedbin.py b/predtuner/pipedbin.py index b8cf1730ba9c199e37f7ebb381e96aa359b439a8..18237d1bb1c5a43109648847ea4cb0c5eb276a54 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),