Skip to content

Commit 1444863

Browse files
authored
Merge pull request #46315 from iarspider/patch-11
Fix testAOTTools for non-amd64 arch
2 parents cc970f2 + d4b344b commit 1444863

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

PhysicsTools/TensorFlowAOT/test/testAOTTools.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
import tempfile
1212
import functools
1313
import unittest
14+
import platform
1415

1516

1617
this_dir = os.path.dirname(os.path.abspath(__file__))
@@ -42,6 +43,8 @@ def test_dev_workflow(self, tmp_dir):
4243
config_file = os.path.join(m.group(1), "share", "test_models", "simple", "aot_config.yaml")
4344
self.assertTrue(os.path.exists(config_file))
4445

46+
arch = "{0}-pc-linux".format(platform.processor())
47+
4548
# run the dev workflow
4649
# create the test model
4750
cmd = [
@@ -50,6 +53,7 @@ def test_dev_workflow(self, tmp_dir):
5053
"-o", tmp_dir,
5154
"--tool-name", "tfaot-model-test",
5255
"--dev",
56+
"--additional-flags=--target_triple=" + arch
5357
]
5458
run_cmd(cmd)
5559

0 commit comments

Comments
 (0)