From 98553f7dffbdc8192fe34c272dc9f763cc630542 Mon Sep 17 00:00:00 2001 From: AwesomeCodingBoy <43309460+ZhangZhiPku@users.noreply.github.com> Date: Wed, 30 Mar 2022 20:37:50 +0800 Subject: [PATCH] Fix Critical Failure of Dispatcher (#43) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 上一次对Dispatcher修改导致参数传递错误,所有Shape算子的调度平台出现错误。这个commit将会修复这些错误。 --- ppq/api/interface.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ppq/api/interface.py b/ppq/api/interface.py index e938b658..da27d208 100644 --- a/ppq/api/interface.py +++ b/ppq/api/interface.py @@ -567,8 +567,8 @@ def dispatch_graph(graph: BaseGraph, platform: TargetPlatform, setting: Quantiza dispatching_table = dispatcher.dispatch( graph=graph, quant_types=quant_types, - quant_platform=quantizer.target_platform, - fp32_platform=quantizer.default_platform, + quant_platform=TargetPlatform.UNSPECIFIED, # MUST BE UNSPECIFIED, 这里的意思是交由 Quantizer 决定是否量化这个算子 + fp32_platform=TargetPlatform.FP32, SOI_platform=TargetPlatform.SHAPE_OR_INDEX) # override dispatching result with setting