Skip to content

Commit 8ec3183

Browse files
SigureMoSigureMo
authored
[Prim][CINN] Use API to instead of set prim all cpp flag (#3051)
1 parent a369b3a commit 8ec3183

File tree

2,969 files changed

+7523
-1451
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

2,969 files changed

+7523
-1451
lines changed

‎framework/e2e/PaddleLT_new/layerE2Ecase/1000-subgraph-ops/all/test_all_builtin_combine.py‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,10 @@ def SetDefaultEnv(**env_var2value):
176176
def SetEnvVar(env_var2value):
177177
for env_var, value in env_var2value.items():
178178
os.environ[env_var] = str(value)
179+
if env_var2value.get("FLAGS_prim_all") is not None:
180+
prim_all_value = env_var2value.pop("FLAGS_prim_all")
181+
env_var2value["FLAGS_prim_forward"] = prim_all_value
182+
env_var2value["FLAGS_prim_backward"] = prim_all_value
179183
paddle.set_flags({
180184
env_var:value
181185
for env_var, value in env_var2value.items()

‎framework/e2e/PaddleLT_new/layerE2Ecase/1000-subgraph-ops/all/test_all_builtin_split.py‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,10 @@ def SetDefaultEnv(**env_var2value):
176176
def SetEnvVar(env_var2value):
177177
for env_var, value in env_var2value.items():
178178
os.environ[env_var] = str(value)
179+
if env_var2value.get("FLAGS_prim_all") is not None:
180+
prim_all_value = env_var2value.pop("FLAGS_prim_all")
181+
env_var2value["FLAGS_prim_forward"] = prim_all_value
182+
env_var2value["FLAGS_prim_backward"] = prim_all_value
179183
paddle.set_flags({
180184
env_var:value
181185
for env_var, value in env_var2value.items()

‎framework/e2e/PaddleLT_new/layerE2Ecase/1000-subgraph-ops/all/test_all_pd_op_abs.py‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,10 @@ def SetDefaultEnv(**env_var2value):
176176
def SetEnvVar(env_var2value):
177177
for env_var, value in env_var2value.items():
178178
os.environ[env_var] = str(value)
179+
if env_var2value.get("FLAGS_prim_all") is not None:
180+
prim_all_value = env_var2value.pop("FLAGS_prim_all")
181+
env_var2value["FLAGS_prim_forward"] = prim_all_value
182+
env_var2value["FLAGS_prim_backward"] = prim_all_value
179183
paddle.set_flags({
180184
env_var:value
181185
for env_var, value in env_var2value.items()

‎framework/e2e/PaddleLT_new/layerE2Ecase/1000-subgraph-ops/all/test_all_pd_op_add.py‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,10 @@ def SetDefaultEnv(**env_var2value):
176176
def SetEnvVar(env_var2value):
177177
for env_var, value in env_var2value.items():
178178
os.environ[env_var] = str(value)
179+
if env_var2value.get("FLAGS_prim_all") is not None:
180+
prim_all_value = env_var2value.pop("FLAGS_prim_all")
181+
env_var2value["FLAGS_prim_forward"] = prim_all_value
182+
env_var2value["FLAGS_prim_backward"] = prim_all_value
179183
paddle.set_flags({
180184
env_var:value
181185
for env_var, value in env_var2value.items()

‎framework/e2e/PaddleLT_new/layerE2Ecase/1000-subgraph-ops/all/test_all_pd_op_add_n.py‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,10 @@ def SetDefaultEnv(**env_var2value):
176176
def SetEnvVar(env_var2value):
177177
for env_var, value in env_var2value.items():
178178
os.environ[env_var] = str(value)
179+
if env_var2value.get("FLAGS_prim_all") is not None:
180+
prim_all_value = env_var2value.pop("FLAGS_prim_all")
181+
env_var2value["FLAGS_prim_forward"] = prim_all_value
182+
env_var2value["FLAGS_prim_backward"] = prim_all_value
179183
paddle.set_flags({
180184
env_var:value
181185
for env_var, value in env_var2value.items()

‎framework/e2e/PaddleLT_new/layerE2Ecase/1000-subgraph-ops/all/test_all_pd_op_all.py‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,10 @@ def SetDefaultEnv(**env_var2value):
176176
def SetEnvVar(env_var2value):
177177
for env_var, value in env_var2value.items():
178178
os.environ[env_var] = str(value)
179+
if env_var2value.get("FLAGS_prim_all") is not None:
180+
prim_all_value = env_var2value.pop("FLAGS_prim_all")
181+
env_var2value["FLAGS_prim_forward"] = prim_all_value
182+
env_var2value["FLAGS_prim_backward"] = prim_all_value
179183
paddle.set_flags({
180184
env_var:value
181185
for env_var, value in env_var2value.items()

‎framework/e2e/PaddleLT_new/layerE2Ecase/1000-subgraph-ops/all/test_all_pd_op_arange.py‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,10 @@ def SetDefaultEnv(**env_var2value):
176176
def SetEnvVar(env_var2value):
177177
for env_var, value in env_var2value.items():
178178
os.environ[env_var] = str(value)
179+
if env_var2value.get("FLAGS_prim_all") is not None:
180+
prim_all_value = env_var2value.pop("FLAGS_prim_all")
181+
env_var2value["FLAGS_prim_forward"] = prim_all_value
182+
env_var2value["FLAGS_prim_backward"] = prim_all_value
179183
paddle.set_flags({
180184
env_var:value
181185
for env_var, value in env_var2value.items()

‎framework/e2e/PaddleLT_new/layerE2Ecase/1000-subgraph-ops/all/test_all_pd_op_argmax.py‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,10 @@ def SetDefaultEnv(**env_var2value):
176176
def SetEnvVar(env_var2value):
177177
for env_var, value in env_var2value.items():
178178
os.environ[env_var] = str(value)
179+
if env_var2value.get("FLAGS_prim_all") is not None:
180+
prim_all_value = env_var2value.pop("FLAGS_prim_all")
181+
env_var2value["FLAGS_prim_forward"] = prim_all_value
182+
env_var2value["FLAGS_prim_backward"] = prim_all_value
179183
paddle.set_flags({
180184
env_var:value
181185
for env_var, value in env_var2value.items()

‎framework/e2e/PaddleLT_new/layerE2Ecase/1000-subgraph-ops/all/test_all_pd_op_assign.py‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,10 @@ def SetDefaultEnv(**env_var2value):
176176
def SetEnvVar(env_var2value):
177177
for env_var, value in env_var2value.items():
178178
os.environ[env_var] = str(value)
179+
if env_var2value.get("FLAGS_prim_all") is not None:
180+
prim_all_value = env_var2value.pop("FLAGS_prim_all")
181+
env_var2value["FLAGS_prim_forward"] = prim_all_value
182+
env_var2value["FLAGS_prim_backward"] = prim_all_value
179183
paddle.set_flags({
180184
env_var:value
181185
for env_var, value in env_var2value.items()

‎framework/e2e/PaddleLT_new/layerE2Ecase/1000-subgraph-ops/all/test_all_pd_op_atan.py‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,10 @@ def SetDefaultEnv(**env_var2value):
176176
def SetEnvVar(env_var2value):
177177
for env_var, value in env_var2value.items():
178178
os.environ[env_var] = str(value)
179+
if env_var2value.get("FLAGS_prim_all") is not None:
180+
prim_all_value = env_var2value.pop("FLAGS_prim_all")
181+
env_var2value["FLAGS_prim_forward"] = prim_all_value
182+
env_var2value["FLAGS_prim_backward"] = prim_all_value
179183
paddle.set_flags({
180184
env_var:value
181185
for env_var, value in env_var2value.items()

0 commit comments

Comments
 (0)