diff --git a/test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh b/test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh index 5146e6ccf65de82ec9e7b44cfb6a2f3f393bb12c..61620aed1d21f5f90597113cb2c9111f52ab19ed 100644 --- a/test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh +++ b/test_tipc/static/ResNet50/benchmark_common/run_benchmark.sh @@ -84,6 +84,7 @@ function _set_env(){ export FLAGS_cudnn_exhaustive_search=1 export FLAGS_eager_delete_tensor_gb=0.0 export FLAGS_conv_workspace_size_limit=4000 + export FLAGS_apply_pass_to_program=true }