Skip to content

Commit

Permalink
Update
Browse files Browse the repository at this point in the history
[ghstack-poisoned]
  • Loading branch information
wconstab committed May 15, 2024
1 parent c1b4c08 commit 7b4c79a
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 7 deletions.
5 changes: 1 addition & 4 deletions .github/workflows/unit_test_4gpu.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ jobs:
gpu-arch-type: cuda
gpu-arch-version: "12.1"
# This image is faster to clone than the default, but it lacks CC needed by triton
# (1m25s vs 2m37s)
# (1m25s vs 2m37s).
docker-image: "pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime"
repository: "pytorch/torchtitan"
upload-artifact: "outputs"
Expand All @@ -32,6 +32,3 @@ jobs:
python -m pip install -r dev-requirements.txt
mkdir artifacts-to-be-uploaded
python ./test_runner.py artifacts-to-be-uploaded
# upload-coverage:
# - name: Upload Coverage to Codecov
# uses: codecov/codecov-action@v3
3 changes: 0 additions & 3 deletions .github/workflows/unit_test_cpu.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,3 @@ jobs:
python -m pip install -r requirements.txt
python -m pip install -r dev-requirements.txt
pytest test --cov=. --cov-report=xml --durations=20 -vv
# upload-coverage:
# - name: Upload Coverage to Codecov
# uses: codecov/codecov-action@v3

0 comments on commit 7b4c79a

Please sign in to comment.