From baa2dba2b2c40aeb2fc7539a1c03d2b18d036ca5 Mon Sep 17 00:00:00 2001 From: mengqinghe0909 Date: Mon, 31 Mar 2025 17:11:25 +0800 Subject: [PATCH] modified 2d_unsteady jit level --- MindFlow/applications/data_driven/airfoil/2D_unsteady/train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/MindFlow/applications/data_driven/airfoil/2D_unsteady/train.py b/MindFlow/applications/data_driven/airfoil/2D_unsteady/train.py index e02dd9473..e9d50dec1 100644 --- a/MindFlow/applications/data_driven/airfoil/2D_unsteady/train.py +++ b/MindFlow/applications/data_driven/airfoil/2D_unsteady/train.py @@ -168,6 +168,7 @@ if __name__ == '__main__': print_log(f"pid: {os.getpid()}") print_log(datetime.datetime.now()) context.set_context(mode=context.GRAPH_MODE if args.mode.upper().startswith("GRAPH") else context.PYNATIVE_MODE, + jit_config={"jit_level": "O2"}, device_target=args.device_target, device_id=args.device_id) print_log(f"device_id: {context.get_context(attr_key='device_id')}") -- Gitee