[workflow] enable testing for develop & feature branch (#3801)

.github/workflows/build_on_pr.yml

@@ -3,24 +3,27 @@ name: Build on PR
 on:
   pull_request:
     types: [synchronize, opened, reopened]
+    branches:
+      - "main"
+      - "develop"
+      - "feature/**"
     paths:
-      - '.github/workflows/build_on_pr.yml' # run command & env variables change
-      - 'colossalai/**' # source code change
-      - '!colossalai/**.md' # ignore doc change
-      - 'op_builder/**' # cuda extension change
-      - '!op_builder/**.md' # ignore doc change
-      - 'requirements/**' # requirements change
-      - 'tests/**' # test change
-      - '!tests/**.md' # ignore doc change
-      - 'pytest.ini' # test config change
-      - 'setup.py' # install command change
+      - ".github/workflows/build_on_pr.yml" # run command & env variables change
+      - "colossalai/**" # source code change
+      - "!colossalai/**.md" # ignore doc change
+      - "op_builder/**" # cuda extension change
+      - "!op_builder/**.md" # ignore doc change
+      - "requirements/**" # requirements change
+      - "tests/**" # test change
+      - "!tests/**.md" # ignore doc change
+      - "pytest.ini" # test config change
+      - "setup.py" # install command change
 
 jobs:
   detect:
     name: Detect file change
     if: |
       github.event.pull_request.draft == false &&
-      github.base_ref == 'main' &&
       github.event.pull_request.base.repo.full_name == 'hpcaitech/ColossalAI'
     outputs:
       changedExtenisonFiles: ${{ steps.find-extension-change.outputs.all_changed_files }}

.github/workflows/doc_check_on_pr.yml

@@ -2,15 +2,18 @@ name: Check Documentation on PR
 
 on:
   pull_request:
+    branches:
+      - "main"
+      - "develop"
+      - "feature/**"
     paths:
-      - 'docs/**'
+      - "docs/**"
 
 jobs:
   check-i18n:
     name: Check docs in diff languages
     if: |
       github.event.pull_request.draft == false &&
-        github.base_ref == 'main' &&
       github.event.pull_request.base.repo.full_name == 'hpcaitech/ColossalAI'
     runs-on: ubuntu-latest
     steps:
@@ -18,7 +21,7 @@ jobs:
 
       - uses: actions/setup-python@v2
         with:
-          python-version: '3.8.14'
+          python-version: "3.8.14"
 
       - run: python .github/workflows/scripts/check_doc_i18n.py -d docs/source
 
@@ -26,23 +29,22 @@ jobs:
     name: Test if the docs can be built
     if: |
       github.event.pull_request.draft == false &&
-        github.base_ref == 'main' &&
       github.event.pull_request.base.repo.full_name == 'hpcaitech/ColossalAI'
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v2
         with:
-          path: './ColossalAI'
+          path: "./ColossalAI"
           fetch-depth: 0
 
       - uses: actions/checkout@v2
         with:
-          path: './ColossalAI-Documentation'
-          repository: 'hpcaitech/ColossalAI-Documentation'
+          path: "./ColossalAI-Documentation"
+          repository: "hpcaitech/ColossalAI-Documentation"
 
       - uses: actions/setup-python@v2
         with:
-          python-version: '3.8.14'
+          python-version: "3.8.14"
 
       # we use the versions in the main branch as the guide for versions to display
       # checkout will give your merged branch
@@ -57,7 +59,6 @@ jobs:
           git config user.name 'github-actions'
           git config user.email 'github-actions@github.com'
 
-
       - name: Build docs
         run: |
           cache_dir=ColossalAI-Documentation/doc-build/.cache

.github/workflows/doc_test_on_pr.yml

 name: Test Documentation on PR
 on:
   pull_request:
+    branches:
+      - "main"
+      - "develop"
+      - "feature/**"
     # any change in the examples folder will trigger check for the corresponding example.
     paths:
-      - 'docs/source/**.md'
+      - "docs/source/**.md"
 
 jobs:
   # This is for changed example files detect and output a matrix containing all the corresponding directory name.
   detect-changed-doc:
     if: |
       github.event.pull_request.draft == false &&
-        github.base_ref == 'main' &&
       github.event.pull_request.base.repo.full_name == 'hpcaitech/ColossalAI' && github.event_name == 'pull_request'
     runs-on: ubuntu-latest
     outputs:
@@ -44,7 +47,6 @@ jobs:
     # Add this condition to avoid executing this job if the trigger event is workflow_dispatch.
     if: |
       github.event.pull_request.draft == false &&
-        github.base_ref == 'main' &&
       github.event.pull_request.base.repo.full_name == 'hpcaitech/ColossalAI' && github.event_name == 'pull_request' &&
       needs.detect-changed-doc.outputs.any_changed == 'true'
     name: Test the changed Doc
@@ -61,8 +63,8 @@ jobs:
       - name: Checkout ColossalAI-Documentation
         uses: actions/checkout@v2
         with:
-          path: './ColossalAI-Documentation'
-          repository: 'hpcaitech/ColossalAI-Documentation'
+          path: "./ColossalAI-Documentation"
+          repository: "hpcaitech/ColossalAI-Documentation"
 
       - name: Install Docer
         run: |

.github/workflows/example_check_on_pr.yml

 name: Test Example on PR
 on:
   pull_request:
+    branches:
+      - "main"
+      - "develop"
+      - "feature/**"
     # any change in the examples folder will trigger check for the corresponding example.
     paths:
-      - 'examples/**'
+      - "examples/**"
 
 jobs:
   # This is for changed example files detect and output a matrix containing all the corresponding directory name.
   detect-changed-example:
     if: |
       github.event.pull_request.draft == false &&
-        github.base_ref == 'main' &&
       github.event.pull_request.base.repo.full_name == 'hpcaitech/ColossalAI' && github.event_name == 'pull_request'
     runs-on: ubuntu-latest
     outputs:
@@ -62,7 +65,6 @@ jobs:
     # Add this condition to avoid executing this job if the trigger event is workflow_dispatch.
     if: |
       github.event.pull_request.draft == false &&
-        github.base_ref == 'main' &&
       github.event.pull_request.base.repo.full_name == 'hpcaitech/ColossalAI' && github.event_name == 'pull_request' &&
       needs.detect-changed-example.outputs.anyChanged == 'true'
     name: Test the changed example