From dbc320182421e7ebac57415386a4c6efb68b3989 Mon Sep 17 00:00:00 2001 From: Xu Zhao Date: Mon, 22 Jan 2024 17:19:31 -0800 Subject: [PATCH] Fix the A10G CI due to permission issues (#2123) Summary: After the version upgrade, the runner UID changed from 1000 to 1001, so we need to setup the directory permission correctly to fix it on A10G CI. Pull Request resolved: https://github.com/pytorch/benchmark/pull/2123 Reviewed By: aaronenyeshi Differential Revision: D52971828 Pulled By: xuzhao9 fbshipit-source-id: 6d63e64b38e2339696578dc3aacd220c9df84681 --- .github/workflows/pr-a10g.yml | 1 + torchbenchmark/models/BERT_pytorch/install.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/pr-a10g.yml b/.github/workflows/pr-a10g.yml index c5f549443d..fb390a7638 100644 --- a/.github/workflows/pr-a10g.yml +++ b/.github/workflows/pr-a10g.yml @@ -52,6 +52,7 @@ jobs: tail -f /dev/null ) echo "Container name: ${container_name}" + docker exec -t -w "/" "${container_name}" bash -c "sudo chown -R runner /benchmark; sudo chgrp -R runner /benchmark" docker exec -t -w "/benchmark" "${container_name}" bash /benchmark/scripts/torchbench_install.sh docker exec -t -w "/benchmark" "${container_name}" bash /benchmark/scripts/torchbench_test.sh - name: Teardown Linux diff --git a/torchbenchmark/models/BERT_pytorch/install.py b/torchbenchmark/models/BERT_pytorch/install.py index 26ab44a808..f1533e07a6 100644 --- a/torchbenchmark/models/BERT_pytorch/install.py +++ b/torchbenchmark/models/BERT_pytorch/install.py @@ -3,7 +3,7 @@ def setup_install(): - subprocess.check_call([sys.executable, 'setup.py', 'develop']) + subprocess.check_call([sys.executable, '-m', "pip", 'install', '--user', '-e', '.']) if __name__ == '__main__': setup_install()