Browse Source

chore(mgblar): delete winograd transform switch

GitOrigin-RevId: aa4051541d
release-1.2
Megvii Engine Team 4 years ago
parent
commit
c3c1e46df7
2 changed files with 0 additions and 12 deletions
  1. +0
    -9
      sdk/load-and-run/src/mgblar.cpp
  2. +0
    -3
      src/core/include/megbrain/graph/cg.h

+ 0
- 9
sdk/load-and-run/src/mgblar.cpp View File

@@ -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();


+ 0
- 3
src/core/include/megbrain/graph/cg.h View File

@@ -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) \


Loading…
Cancel
Save