25
25
TEST_TYPE : " single_gpu_${{ matrix.docker-image-name }}"
26
26
container :
27
27
image : ${{ matrix.docker-image-name }}
28
- options : --gpus all --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/ -e NVIDIA_DISABLE_REQUIRE=true
28
+ options : --gpus all --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
29
29
defaults :
30
30
run :
31
31
shell : bash
@@ -39,16 +39,19 @@ jobs:
39
39
mkdir transformers-clone && git clone https://github.com/huggingface/transformers.git transformers-clone # rename to transformers clone to avoid modules conflict
40
40
41
41
- name : Run examples on single GPU
42
+ if : always()
42
43
run : |
43
44
source activate peft
44
45
make tests_examples_single_gpu_bnb
45
46
46
47
- name : Run core tests on single GPU
48
+ if : always()
47
49
run : |
48
50
source activate peft
49
51
make tests_core_single_gpu_bnb
50
52
51
53
- name : Run transformers tests on single GPU
54
+ if : always()
52
55
run : |
53
56
source activate peft
54
57
make transformers_tests
70
73
TEST_TYPE : " multi_gpu_${{ matrix.docker-image-name }}"
71
74
container :
72
75
image : ${{ matrix.docker-image-name }}
73
- options : --gpus all --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/ -e NVIDIA_DISABLE_REQUIRE=true
76
+ options : --gpus all --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
74
77
defaults :
75
78
run :
76
79
shell : bash
@@ -84,20 +87,24 @@ jobs:
84
87
mkdir transformers-clone && git clone https://github.com/huggingface/transformers.git transformers-clone
85
88
86
89
- name : Run core GPU tests on multi-gpu
90
+ if : always()
87
91
run : |
88
92
source activate peft
89
93
90
94
- name : Run examples on multi GPU
95
+ if : always()
91
96
run : |
92
97
source activate peft
93
98
make tests_examples_multi_gpu_bnb
94
99
95
100
- name : Run core tests on multi GPU
101
+ if : always()
96
102
run : |
97
103
source activate peft
98
104
make tests_core_multi_gpu_bnb
99
105
100
106
- name : Run transformers tests on multi GPU
107
+ if : always()
101
108
run : |
102
109
source activate peft
103
110
make transformers_tests
0 commit comments