From 78fe9e55a30239ab98d5eb5b002b4620d39d7d74 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Mon, 11 May 2020 22:19:01 +0800 Subject: [PATCH] feat(mge): add more options for jit dump GitOrigin-RevId: c56d42807c110d97e843045def4d960574ad945a --- python_module/megengine/jit/__init__.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python_module/megengine/jit/__init__.py b/python_module/megengine/jit/__init__.py index 7da88c915..89b134473 100644 --- a/python_module/megengine/jit/__init__.py +++ b/python_module/megengine/jit/__init__.py @@ -464,8 +464,10 @@ class trace: "enable_ioc16": "f16_io_comp", "enable_hwcd4": "use_nhwcd4", "enable_nchw88": "use_nchw88", + "enable_nchw32": "use_nchw32", + "enable_nchw44": "use_nchw44", + "enable_chwn4": "use_chwn4", "enable_fuse_conv_bias_nonlinearity": "fuse_conv_bias_nonlinearity", - "enable_tensorcore": "use_tensor_core", "enable_fuse_conv_bias_with_z": "fuse_conv_bias_with_z", } if optimize_for_inference: -- GitLab