add

mosaicml · Oct 9, 2023 · e80ef3e · e80ef3e
1 parent d1f8773
commit e80ef3e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/.github/mcp/mcp_pytest.py b/.github/mcp/mcp_pytest.py
@@ -95,6 +95,8 @@
 
     make test PYTEST='{args.pytest_command}' EXTRA_ARGS="$COMMON_ARGS --codeblocks"
 
+    pkill -9 python
+
     make test-dist PYTEST='{args.pytest_command}' EXTRA_ARGS="$COMMON_ARGS" WORLD_SIZE=2
 
     python -m coverage combine

diff --git a/.github/workflows/pr-gpu.yaml b/.github/workflows/pr-gpu.yaml
@@ -19,6 +19,10 @@ jobs:
     strategy:
       matrix:
         include:
+        - name: 'gpu-latest-aws'
+          container: mosaicml/pytorch:latest-aws
+          markers: 'gpu'
+          pytest_command: 'coverage run -m pytest'
         - name: 'gpu-latest'
           container: mosaicml/pytorch:latest  # mosaicml/pytorch:1.13.1_cu117-python3.10-ubuntu20.04
           markers: 'gpu'
@@ -31,6 +35,10 @@ jobs:
           container: mosaicml/pytorch:2.1.0_cu121-python3.10-ubuntu20.04
           markers: 'gpu'
           pytest_command: 'coverage run -m pytest'
+        - name: 'gpu-2.1.0-aws'
+          container: mosaicml/pytorch:2.1.0_cu121-python3.10-ubuntu20.04-aws
+          markers: 'gpu'
+          pytest_command: 'coverage run -m pytest'
     name: ${{ matrix.name }}
     if: github.repository_owner == 'mosaicml'
     with: