提交 c3c1e46d 编写于 作者: M Megvii Engine Team

chore(mgblar): delete winograd transform switch

GitOrigin-RevId: aa4051541ddfe7ff0cb0ba8fa50526acbc919fb1
上级 82847bc7
...@@ -190,10 +190,6 @@ R"__usage__( ...@@ -190,10 +190,6 @@ R"__usage__(
--enable-jit --enable-jit
Execute supported operators with JIT(now only support NVRTC). Can only be used on Nvidia GPUs. Execute supported operators with JIT(now only support NVRTC). Can only be used on Nvidia GPUs.
)__usage__" )__usage__"
R"__usage__(
--winograd-transform
Execute opr replace, replace weights by winograd transform. Currently support on conv bias opr.
)__usage__"
R"__usage__( R"__usage__(
--enable-chwn4 --enable-chwn4
Execute operators with kernels implemented in MegDNN with CHWN4 tensor format. Can only be used Execute operators with kernels implemented in MegDNN with CHWN4 tensor format. Can only be used
...@@ -1391,11 +1387,6 @@ Args Args::from_argv(int argc, char **argv) { ...@@ -1391,11 +1387,6 @@ Args Args::from_argv(int argc, char **argv) {
graph_opt.graph_opt.jit = 1; graph_opt.graph_opt.jit = 1;
continue; continue;
} }
if (!strcmp(argv[i], "--winograd-transform")) {
mgb_log_warn("enable winograd transform");
graph_opt.graph_opt.weight_winograd_transform = true;
continue;
}
if (!strcmp(argv[i], "--weight-preprocess")) { if (!strcmp(argv[i], "--weight-preprocess")) {
mgb_log_warn("enable weight-preprocess optimization"); mgb_log_warn("enable weight-preprocess optimization");
graph_opt.graph_opt.enable_weight_preprocess(); graph_opt.graph_opt.enable_weight_preprocess();
......
...@@ -95,8 +95,6 @@ struct GraphCommonOptimizeOptions { ...@@ -95,8 +95,6 @@ struct GraphCommonOptimizeOptions {
//! fuse pattern like ReLU(conv_bias(x, w, b) + z) or conv_bias(x, w, b) //! fuse pattern like ReLU(conv_bias(x, w, b) + z) or conv_bias(x, w, b)
//! + z -> conv_bias(x, w, b, z) //! + z -> conv_bias(x, w, b, z)
bool fuse_conv_bias_with_z = false; bool fuse_conv_bias_with_z = false;
//! whether to enable fast-run profiled winograd opr replace
bool weight_winograd_transform = false;
//! whether to enable weight preprocess, if enabled it may use more //! whether to enable weight preprocess, if enabled it may use more
//! memory, default disable now, when weight preprocess is enabled, the //! memory, default disable now, when weight preprocess is enabled, the
//! input shape should no change //! input shape should no change
...@@ -133,7 +131,6 @@ struct GraphCommonOptimizeOptions { ...@@ -133,7 +131,6 @@ struct GraphCommonOptimizeOptions {
SET(fuse_conv_bias_nonlinearity); SET(fuse_conv_bias_nonlinearity);
SET(fuse_conv_bias_with_z); SET(fuse_conv_bias_with_z);
SET(fuse_preprocess); SET(fuse_preprocess);
SET(weight_winograd_transform);
SET(weight_preprocess); SET(weight_preprocess);
#undef SET #undef SET
#define SET(_trans, _trans_capital) \ #define SET(_trans, _trans_capital) \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册