From 6474e31556ae011410f29d8d0a2d80de67ed6956 Mon Sep 17 00:00:00 2001
From: Frank Lee <somerlee.9@gmail.com>
Date: Fri, 12 Aug 2022 17:06:29 +0800
Subject: [PATCH] [workflow] added TensorNVMe to compatibility test (#1449)

---
 .github/workflows/compatibility_test.yml | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/.github/workflows/compatibility_test.yml b/.github/workflows/compatibility_test.yml
index acc84edbe..7948eb20c 100644
--- 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