fix test

uwsampl · Nov 20, 2022 · a47a0eb · a47a0eb
1 parent 289cf1d
commit a47a0eb
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 30 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -12,6 +12,8 @@ jobs:
 
   build-docker-image:
     runs-on: self-hosted
+    outputs:
+      tag: ${{ steps.generate-tag.outputs.tag }}
     steps:
     - uses: actions/checkout@v3
       with:
@@ -21,8 +23,7 @@ jobs:
       id: generate-tag
       run: |
         export TAG=sparsetir:$(date +%s)
-        echo "Docker image tag: $TAG"
-        echo "::set-output name=tag::$TAG"
+        echo "tag=$TAG" >> $GITHUB_OUTPUT
     
     - name: Build Docker image
       run: |
@@ -36,13 +37,26 @@ jobs:
     needs: build-docker-image
     steps:
     - name: Run tests
-      run: docker run --gpus all ${{ needs.build-docker-image.outputs.tag }} bash tests/scripts/task_python_sparsetir_unittest.sh
+      id: run-tests
+      run: docker run --gpus all ${{ needs.outputs.tag }} bash tests/scripts/task_python_sparsetir_unittest.sh
 
-  build-doc-and-push-gh-pages:
+  build-doc:
     runs-on: self-hosted
     needs: build-docker-image
     steps:
     - name: Build Documentation
-      run: docker run ${{ needs.build-docker-image.outputs.tag }} sphinx-build -b html docs/ docs-html/
-    # - name: Push changes
-    #   run: docker 
+      id: build-doc
+      run: |
+        docker run ${{ needs.outputs.tag }} sphinx-build -b html docs/ public/
+        docker cp ${{ needs.outputs.tag }}:public/ public/
+  
+  deploy-doc:
+    runs-on: self-hosted
+    needs: build-doc
+    steps:
+    - name: Deploy documentation
+      id: deploy-doc
+      uses: peaceiris/actions-gh-pages@v3
+      with:
+        github_token: ${{ secrets.GITHUB_TOKEN }}
+        publish_dir: public
diff --git a/.github/workflows/docs.yml b/.github/workflows/docs.yml
diff --git a/README.md b/README.md
@@ -22,7 +22,6 @@ SparseTIR: Sparse Tensor Compiler for Deep Learning
 [Paper](https://arxiv.org/abs/2207.04606)
 
 [![Build Status](https://github.com/uwsampl/sparsetir/actions/workflows/build.yml/badge.svg)](https://github.com/uwsampl/sparsetir/actions/workflows/build.yml)
-[![Documentation Status](https://github.com/uwsampl/sparsetir/actions/workflows/docs.yml/badge.svg)](https://github.com/uwsampl/sparsetir/actions/workflows/docs.yml)
 
 SparseTIR is a tensor-level compiler for sparse/irregular operators in Deep Learning. The design goal of SparseTIR is to provide a general programming abstraction that can cover both sparse and irregular (e.g. Ragged Tensors) workloads in Deep Learning including Graph Neural Networks, Sparse Transformers, Sparse Convolutions, Network Pruning, etc. while generating high-performance code on heterogeneous hardware.