From 6c072a78274d38de97744f5e2928e734cc392822 Mon Sep 17 00:00:00 2001 From: clackhan Date: Tue, 12 Mar 2024 16:27:41 +0800 Subject: [PATCH] open_conv_bias_add_act_fusion_default --- src/onediff/infer_compiler/oneflow_compiler_config.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/onediff/infer_compiler/oneflow_compiler_config.py b/src/onediff/infer_compiler/oneflow_compiler_config.py index 287216591..087d38a63 100644 --- a/src/onediff/infer_compiler/oneflow_compiler_config.py +++ b/src/onediff/infer_compiler/oneflow_compiler_config.py @@ -28,6 +28,7 @@ def init_default_env(): os.environ.setdefault("ONEFLOW_KERNEL_GEMM_CUTLASS_IMPL_ENABLE_TUNING_WARMUP", "1") os.environ.setdefault("ONEFLOW_KERNEL_CONV_ENABLE_CUTLASS_IMPL", "1") os.environ.setdefault("ONEFLOW_KERNEL_GEMM_ENABLE_CUTLASS_IMPL", "1") + os.environ.setdefault("ONEFLOW_CONVOLUTION_BIAS_ADD_ACT_FUSION", "1") # os.environ.setdefault("ONEFLOW_KERNEL_GLU_ENABLE_DUAL_GEMM_IMPL", "0") # os.environ.setdefault("ONEFLOW_KERNEL_GLU_ENABLE_Y_GEMM_IMPL", "0") # os.environ.setdefault("ONEFLOW_KERNEL_GLU_QUANT_ENABLE_DUAL_GEMM_IMPL", "0")