trim trailing whitespaces (#3048)

Author: SigureMo

framework/e2e/PaddleLT_new/engine/paddle_eval.py

@@ -152,10 +152,7 @@ def dy2st_eval_cinn(self):
             Logger("dy2st_eval_cinn").get_log().info("已设定环境变量FLAGS_enable_auto_recompute")
         data = self._net_input()
         net = self._net_instant()
-
-        build_strategy = paddle.static.BuildStrategy()
-        build_strategy.build_cinn_pass = True
-        cinn_net = paddle.jit.to_static(net, build_strategy=build_strategy, full_graph=True)
+        cinn_net = paddle.jit.to_static(net, backend="CINN", full_graph=True)
         cinn_net.eval()
         logit = cinn_net(*data)
 
@@ -175,9 +172,7 @@ def dy2st_eval_cinn_inputspec(self):
         Logger("dy2st_eval_cinn_inputspec").get_log().info(f"待测动态InputSpec为: {input_spec}")
         net = self._net_instant()
 
-        build_strategy = paddle.static.BuildStrategy()
-        build_strategy.build_cinn_pass = True
-        cinn_net = paddle.jit.to_static(net, build_strategy=build_strategy, full_graph=True, input_spec=input_spec)
+        cinn_net = paddle.jit.to_static(net, backend="CINN", full_graph=True, input_spec=input_spec)
         cinn_net.eval()
         logit = cinn_net(*data)
 
@@ -197,9 +192,7 @@ def dy2st_eval_cinn_static_inputspec(self):
         Logger("dy2st_eval_cinn_static_inputspec").get_log().info(f"待测静态InputSpec为: {input_spec}")
         net = self._net_instant()
 
-        build_strategy = paddle.static.BuildStrategy()
-        build_strategy.build_cinn_pass = True
-        cinn_net = paddle.jit.to_static(net, build_strategy=build_strategy, full_graph=True, input_spec=input_spec)
+        cinn_net = paddle.jit.to_static(net, backend="CINN", full_graph=True, input_spec=input_spec)
         cinn_net.eval()
         logit = cinn_net(*data)
 

framework/e2e/PaddleLT_new/engine/paddle_eval_bm.py

@@ -151,10 +151,7 @@ def _perf(input_data):
 
     def _dy2st_eval_cinn_perf(self, perf_repeat=10):
         net = self._net_instant()
-
-        build_strategy = paddle.static.BuildStrategy()
-        build_strategy.build_cinn_pass = True
-        cinn_net = paddle.jit.to_static(net, build_strategy=build_strategy, full_graph=True)
+        cinn_net = paddle.jit.to_static(net, backend="CINN", full_graph=True)
         cinn_net.eval()
 
         # net.eval()

framework/e2e/PaddleLT_new/engine/paddle_export.py

@@ -116,10 +116,7 @@ def jit_save_cinn(self):
         """jit.save(layer)"""
         data = self._net_input()
         net = self._net_instant()
-
-        build_strategy = paddle.static.BuildStrategy()
-        build_strategy.build_cinn_pass = True
-        cinn_net = paddle.jit.to_static(net, build_strategy=build_strategy, full_graph=True)
+        cinn_net = paddle.jit.to_static(net, backend="CINN", full_graph=True)
         cinn_net.eval()
         cinn_net(*data)
 
@@ -133,8 +130,6 @@ def jit_save_cinn_inputspec(self):
         Logger("jit_save_cinn_inputspec").get_log().info(f"待测动态InputSpec为: {input_spec}")
         net = self._net_instant()
 
-        build_strategy = paddle.static.BuildStrategy()
-        build_strategy.build_cinn_pass = True
         cinn_net = paddle.jit.to_static(net, full_graph=True, input_spec=input_spec)
         cinn_net.eval()
         # cinn_net(*self._net_input())
@@ -149,8 +144,6 @@ def jit_save_cinn_static_inputspec(self):
         Logger("jit_save_cinn_static_inputspec").get_log().info(f"待测静态InputSpec为: {input_spec}")
         net = self._net_instant()
 
-        build_strategy = paddle.static.BuildStrategy()
-        build_strategy.build_cinn_pass = True
         cinn_net = paddle.jit.to_static(net, full_graph=True, input_spec=input_spec)
         cinn_net.eval()
         # cinn_net(*self._net_input())

framework/e2e/PaddleLT_new/engine/paddle_train.py

@@ -327,10 +327,7 @@ def dy2st_train_cinn(self):
         loss = self._net_loss()
 
         net.train()
-        build_strategy = paddle.static.BuildStrategy()
-        build_strategy.build_cinn_pass = True
-        cinn_net = paddle.jit.to_static(net, build_strategy=build_strategy, full_graph=True)
-
+        cinn_net = paddle.jit.to_static(net, backend="CINN", full_graph=True)
         # 构建optimizer用于训练
         if cinn_net.parameters():
             opt = optimizer.get_opt(net=cinn_net)
@@ -366,9 +363,7 @@ def dy2st_train_cinn_inputspec(self):
         loss = self._net_loss()
 
         net.train()
-        build_strategy = paddle.static.BuildStrategy()
-        build_strategy.build_cinn_pass = True
-        cinn_net = paddle.jit.to_static(net, build_strategy=build_strategy, full_graph=True, input_spec=input_spec)
+        cinn_net = paddle.jit.to_static(net, backend="CINN", full_graph=True, input_spec=input_spec)
 
         # 构建optimizer用于训练
         if cinn_net.parameters():
@@ -405,9 +400,7 @@ def dy2st_train_cinn_static_inputspec(self):
         loss = self._net_loss()
 
         net.train()
-        build_strategy = paddle.static.BuildStrategy()
-        build_strategy.build_cinn_pass = True
-        cinn_net = paddle.jit.to_static(net, build_strategy=build_strategy, full_graph=True, input_spec=input_spec)
+        cinn_net = paddle.jit.to_static(net, backend="CINN", full_graph=True, input_spec=input_spec)
 
         # 构建optimizer用于训练
         if cinn_net.parameters():

framework/e2e/PaddleLT_new/engine/paddle_train_bm.py

@@ -204,10 +204,7 @@ def _dy2st_train_cinn_perf(self, perf_repeat=10):
         loss = self._net_loss()
 
         net.train()
-        build_strategy = paddle.static.BuildStrategy()
-        build_strategy.build_cinn_pass = True
-        cinn_net = paddle.jit.to_static(net, build_strategy=build_strategy, full_graph=True)
-
+        cinn_net = paddle.jit.to_static(net, backend="CINN", full_graph=True)
         # 构建optimizer用于训练
         if cinn_net.parameters():
             opt = optimizer.get_opt(net=cinn_net)