[workflow] added TensorNVMe to compatibility test (hpcaitech#1449)

widyhu2020 · Aug 12, 2022 · 6474e31 · 6474e31
1 parent 9f3eed6
commit 6474e31
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/.github/workflows/compatibility_test.yml b/.github/workflows/compatibility_test.yml
@@ -57,6 +57,19 @@ jobs:
  run: |
  pip install -U pip setuptools wheel --user
  - uses: actions/checkout@v2
+ with:
+ repository: hpcaitech/TensorNVMe
+ ssh-key: ${{ secrets.SSH_KEY_FOR_CI }}
+ path: TensorNVMe
+ - name: Install tensornvme
+ run: |
+ cd TensorNVMe
+ conda install cmake
+ pip install -r requirements.txt
+ pip install -v .
+ - uses: actions/checkout@v2
+ with:
+ ssh-key: ${{ secrets.SSH_KEY_FOR_CI }}
  - name: Install Colossal-AI 
  run: |
  pip install -r requirements/requirements.txt
@@ -68,3 +81,4 @@ jobs:
  env:
  DATA: /data/scratch/cifar-10
  NCCL_SHM_DISABLE: 1
+ LD_LIBRARY_PATH: /github/home/.tensornvme/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64