[PATCH v2 0/1][autotest-client-tests] create deep learning performance test

Taihsiang Ho (tai271828) taihsiang.ho at canonical.com
Mon Jul 5 14:48:09 UTC 2021


Create the first job set for deep learning framework performance test.
Currently only TensorFlow is supported.

Test this patch by:
1. Grab DGX-1, DGX-2, DGX-A100, or any machine with enabled GPUs or other devices supported by the target TensorFlow version.
2. Setup autotest as usual.
3. TEST_CONFIG=config1 AUTOTEST_PATH=/home/ubuntu/autotest sudo -E autotest/client/autotest-local --verbose autotest/client/tests/ubuntu_performance_deep_learning/control

Taihsiang Ho (tai271828) (1):
  UBUNTU: SAUCE: ubuntu_performance_deep_learning: init deep learning
    framework performance test

 ubuntu_performance_deep_learning/control      | 12 +++
 ubuntu_performance_deep_learning/helper.py    | 27 ++++++
 .../ubuntu_performance_deep_learning.py       | 91 +++++++++++++++++++
 .../ubuntu_performance_tensor_flow.sh         | 69 ++++++++++++++
 4 files changed, 199 insertions(+)
 create mode 100644 ubuntu_performance_deep_learning/control
 create mode 100644 ubuntu_performance_deep_learning/helper.py
 create mode 100644 ubuntu_performance_deep_learning/ubuntu_performance_deep_learning.py
 create mode 100755 ubuntu_performance_deep_learning/ubuntu_performance_tensor_flow.sh

-- 
2.32.0




More information about the kernel-team mailing list