FilesExpand file tree

flashpack

/

pr_test_fetcher.yml

Copy path

View runs

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

170 lines (151 loc) · 5.03 KB

flashpack

/

pr_test_fetcher.yml

Copy path

Top

File metadata and controls

• Code
• Blame

170 lines (151 loc) · 5.03 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

name: Fast tests for PRs - Test Fetcher

on: workflow_dispatch

env:

DIFFUSERS_IS_CI: yes

OMP_NUM_THREADS: 4

MKL_NUM_THREADS: 4

PYTEST_TIMEOUT: 60

concurrency:

group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}

cancel-in-progress: true

jobs:

setup_pr_tests:

name: Setup PR Tests

runs-on:

group: aws-general-8-plus

container:

image: diffusers/diffusers-pytorch-cpu

options: --shm-size "16gb" --ipc host -v /mnt/hf_cache:/mnt/cache/

defaults:

run:

shell: bash

outputs:

matrix: ${{ steps.set_matrix.outputs.matrix }}

test_map: ${{ steps.set_matrix.outputs.test_map }}

steps:

- name: Checkout diffusers

uses: actions/checkout@v6

with:

fetch-depth: 0

- name: Install dependencies

run: |

uv pip install -e ".[quality]"

- name: Environment

run: |

python utils/print_env.py

echo $(git --version)

- name: Fetch Tests

run: |

python utils/tests_fetcher.py | tee test_preparation.txt

- name: Report fetched tests

uses: actions/upload-artifact@v6

with:

name: test_fetched

path: test_preparation.txt

- id: set_matrix

name: Create Test Matrix

# The `keys` is used as GitHub actions matrix for jobs, i.e. `models`, `pipelines`, etc.

# The `test_map` is used to get the actual identified test files under each key.

# If no test to run (so no `test_map.json` file), create a dummy map (empty matrix will fail)

run: |

if [ -f test_map.json ]; then

keys=$(python3 -c 'import json; fp = open("test_map.json"); test_map = json.load(fp); fp.close(); d = list(test_map.keys()); print(json.dumps(d))')

test_map=$(python3 -c 'import json; fp = open("test_map.json"); test_map = json.load(fp); fp.close(); print(json.dumps(test_map))')

else

keys=$(python3 -c 'keys = ["dummy"]; print(keys)')

test_map=$(python3 -c 'test_map = {"dummy": []}; print(test_map)')

fi

echo $keys

echo $test_map

echo "matrix=$keys" >> $GITHUB_OUTPUT

echo "test_map=$test_map" >> $GITHUB_OUTPUT

run_pr_tests:

name: Run PR Tests

needs: setup_pr_tests

if: contains(fromJson(needs.setup_pr_tests.outputs.matrix), 'dummy') != true

strategy:

fail-fast: false

max-parallel: 2

matrix:

modules: ${{ fromJson(needs.setup_pr_tests.outputs.matrix) }}

runs-on:

group: aws-general-8-plus

container:

image: diffusers/diffusers-pytorch-cpu

options: --shm-size "16gb" --ipc host -v /mnt/hf_cache:/mnt/cache/

defaults:

run:

shell: bash

steps:

- name: Checkout diffusers

uses: actions/checkout@v6

with:

fetch-depth: 2

- name: Install dependencies

run: |

uv pip install -e ".[quality]"

uv pip install accelerate

- name: Environment

run: |

python utils/print_env.py

- name: Run all selected tests on CPU

run: |

pytest -n 2 --dist=loadfile -v --make-reports=${{ matrix.modules }}_tests_cpu ${{ fromJson(needs.setup_pr_tests.outputs.test_map)[matrix.modules] }}

- name: Failure short reports

if: ${{ failure() }}

continue-on-error: true

run: |

cat reports/${{ matrix.modules }}_tests_cpu_stats.txt

cat reports/${{ matrix.modules }}_tests_cpu_failures_short.txt

- name: Test suite reports artifacts

if: ${{ always() }}

uses: actions/upload-artifact@v6

with:

name: ${{ matrix.modules }}_test_reports

path: reports

run_staging_tests:

strategy:

fail-fast: false

matrix:

config:

- name: Hub tests for models, schedulers, and pipelines

framework: hub_tests_pytorch

runner: aws-general-8-plus

image: diffusers/diffusers-pytorch-cpu

report: torch_hub

name: ${{ matrix.config.name }}

runs-on:

group: ${{ matrix.config.runner }}

container:

image: ${{ matrix.config.image }}

options: --shm-size "16gb" --ipc host -v /mnt/hf_cache:/mnt/cache/

defaults:

run:

shell: bash

steps:

- name: Checkout diffusers

uses: actions/checkout@v6

with:

fetch-depth: 2

- name: Install dependencies

run: |

pip install -e [quality]

- name: Environment

run: |

python utils/print_env.py

- name: Run Hub tests for models, schedulers, and pipelines on a staging env

if: ${{ matrix.config.framework == 'hub_tests_pytorch' }}

run: |

HUGGINGFACE_CO_STAGING=true pytest \

-m "is_staging_test" \

--make-reports=tests_${{ matrix.config.report }} \

tests

- name: Failure short reports

if: ${{ failure() }}

run: cat reports/tests_${{ matrix.config.report }}_failures_short.txt

- name: Test suite reports artifacts

if: ${{ always() }}

uses: actions/upload-artifact@v6

with:

name: pr_${{ matrix.config.report }}_test_reports

path: reports