GitOrigin-RevId: aa4051541d
release-1.2
@@ -191,10 +191,6 @@ R"__usage__( | |||||
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__( | R"__usage__( | ||||
--winograd-transform | |||||
Execute opr replace, replace weights by winograd transform. Currently support on conv bias opr. | |||||
)__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 | ||||
on Nvidia GPUs, whose compute capability is above 6.1. | on Nvidia GPUs, whose compute capability is above 6.1. | ||||
@@ -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 { | |||||
//! 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 { | |||||
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) \ | ||||