@@ -77,23 +77,13 @@ def pytest_addoption(parser):
77
77
"--llama3-8b-tokenizer-path" ,
78
78
type = Path ,
79
79
action = "store" ,
80
- default = "/data/llama-3.1/8b/tokenizer_config.json" ,
81
80
help = "Llama3.1 8b tokenizer path, defaults to 30F CI system path" ,
82
81
)
83
82
84
- parser .addoption (
85
- "--llama3-8b-json-path" ,
86
- type = Path ,
87
- action = "store" ,
88
- default = "/data/extra/models/llama3.1_8B/llama8b_test.json" ,
89
- help = "Llama3.1 8b fp8 parameters json path" ,
90
- )
91
-
92
83
parser .addoption (
93
84
"--llama3-8b-f16-model-path" ,
94
85
type = Path ,
95
86
action = "store" ,
96
- default = "/data/llama-3.1/8b/llama8b_f16.irpa" ,
97
87
help = "Llama3.1 8b model path, defaults to 30F CI system path" ,
98
88
)
99
89
@@ -105,51 +95,17 @@ def pytest_addoption(parser):
105
95
help = "Llama3.1 8b fp8 model path" ,
106
96
)
107
97
108
- parser .addoption (
109
- "--llama3-8b-f16-mlir-path" ,
110
- type = Path ,
111
- action = "store" ,
112
- default = "/data/extra/models/llama3.1_8B/llama8b_f16_test.mlir" ,
113
- help = "Llama3.1 8b mlir path, defaults to 30F CI system path" ,
114
- )
115
-
116
- parser .addoption (
117
- "--llama3-8b-fp8-mlir-path" ,
118
- type = Path ,
119
- action = "store" ,
120
- default = None ,
121
- help = "Llama3.1 8b fp8 mlir path" ,
122
- )
123
-
124
- parser .addoption (
125
- "--llama3-8b-f16-vmfb-path" ,
126
- type = Path ,
127
- action = "store" ,
128
- default = "/data/extra/models/llama3.1_8B/llama8b_f16.vmfb" ,
129
- help = "Llama3.1 8b fp16 vmfb path, defaults to 30F CI system path" ,
130
- )
131
-
132
98
parser .addoption (
133
99
"--llama3-405b-tokenizer-path" ,
134
100
type = Path ,
135
101
action = "store" ,
136
- default = "/data/llama-3.1/405b/tokenizer_config.json" ,
137
102
help = "Llama3.1 405b tokenizer path, defaults to 30F CI system path" ,
138
103
)
139
104
140
- parser .addoption (
141
- "--llama3-405b-json-path" ,
142
- type = Path ,
143
- action = "store" ,
144
- default = "/data/extra/models/llama3.1_405B/llama405b_test.json" ,
145
- help = "Llama3.1 405b fp8 parameters json path" ,
146
- )
147
-
148
105
parser .addoption (
149
106
"--llama3-405b-f16-model-path" ,
150
107
type = Path ,
151
108
action = "store" ,
152
- default = "/data/llama-3.1/405b/llama405b_fp16.irpa" ,
153
109
help = "Llama3.1 405b model path, defaults to 30F CI system path" ,
154
110
)
155
111
@@ -161,30 +117,6 @@ def pytest_addoption(parser):
161
117
help = "Llama3.1 405b fp8 model path" ,
162
118
)
163
119
164
- parser .addoption (
165
- "--llama3-405b-f16-mlir-path" ,
166
- type = Path ,
167
- action = "store" ,
168
- default = "/data/extra/models/llama3.1_405B/llama405b_fp16_test.mlir" ,
169
- help = "Llama3.1 405b mlir path, defaults to 30F CI system path" ,
170
- )
171
-
172
- parser .addoption (
173
- "--llama3-405b-fp8-mlir-path" ,
174
- type = Path ,
175
- action = "store" ,
176
- default = None ,
177
- help = "Llama3.1 405b fp8 mlir path" ,
178
- )
179
-
180
- parser .addoption (
181
- "--llama3-405b-f16-vmfb-path" ,
182
- type = Path ,
183
- action = "store" ,
184
- default = "/data/extra/models/llama3.1_405B/llama405b_fp16.vmfb" ,
185
- help = "Llama3.1 405b fp16 vmfb path, defaults to 30F CI system path" ,
186
- )
187
-
188
120
parser .addoption (
189
121
"--baseline-perplexity-scores" ,
190
122
type = Path ,
@@ -203,7 +135,6 @@ def pytest_addoption(parser):
203
135
parser .addoption (
204
136
"--iree-hip-target" ,
205
137
action = "store" ,
206
- default = "gfx942" ,
207
138
help = "Specify the iree-hip target version (e.g., gfx942)" ,
208
139
)
209
140
0 commit comments