Skip to content

Commit be8f7df

Browse files
Merge branch 'master' into feature/world_size_getter
2 parents 05c42e0 + 64ac13f commit be8f7df

File tree

3 files changed

+9
-6
lines changed

3 files changed

+9
-6
lines changed

.github/workflows/modal-accelerate.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ name: modal-accelerate
33
# This CI is running on modal.com's GPUs.
44
#
55
# It's set up here on github actions and then the cloned repo is sent to modal and everything
6-
# happens on their hw - see deepspeed/modal_ci/accelerate.py for where the actual vm is loaded, updated and the tests are
6+
# happens on their hw - see ci/accelerate.py for where the actual vm is loaded, updated and the tests are
77
# run.
88
#
99
# Both files are annotated to what's important and how one might change or update things if needed.
@@ -20,13 +20,13 @@ on:
2020
branches:
2121
- master
2222

23-
pull_request:
23+
pull_request_target:
2424
paths-ignore:
2525
- 'docs/**'
2626
- 'blogs/**'
2727
- 'deepspeed/inference/v2/**'
2828
- 'tests/unit/inference/v2/**'
29-
types: [draft, opened, ready_for_review, synchronize]
29+
types: [review_requested, ready_for_review, synchronize]
3030
branches:
3131
- master
3232

.github/workflows/modal-torch-latest.yml

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ name: modal-torch-latest
33
# This CI is running on modal.com's GPUs.
44
#
55
# It's set up here on github actions and then the cloned repo is sent to modal and everything
6-
# happens on their hw - see deepspeed/modal_ci/torch_latest.py for where the actual vm is loaded, updated and the tests are
6+
# happens on their hw - see ci/torch_latest.py for where the actual vm is loaded, updated and the tests are
77
# run.
88
#
99
# Both files are annotated to what's important and how one might change or update things if needed.
@@ -16,17 +16,18 @@ name: modal-torch-latest
1616

1717
on:
1818
workflow_dispatch:
19+
1920
push:
2021
branches:
2122
- master
2223

23-
pull_request:
24+
pull_request_target:
2425
paths-ignore:
2526
- 'docs/**'
2627
- 'blogs/**'
2728
- 'deepspeed/inference/v2/**'
2829
- 'tests/unit/inference/v2/**'
29-
types: [draft, opened, ready_for_review, synchronize]
30+
types: [review_requested, ready_for_review, synchronize]
3031
branches:
3132
- master
3233

accelerator/cuda_accelerator.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -245,6 +245,8 @@ def communication_backend_name(self):
245245
return self._communication_backend_name
246246

247247
def is_triton_supported(self):
248+
if not self.is_available():
249+
return False
248250
major, _ = torch.cuda.get_device_capability()
249251
if major >= 8:
250252
return True

0 commit comments

Comments
 (0)