Google Git
Sign in
opensecura / 3p / openxla / iree / 6145b65a80e67f084885215dd75280c7ea1d4daf / . / tests / e2e / stablehlo_ops / CMakeLists.txt
blob: 2050c5286702a23edf7b49e804e732a25c6e9e94 [file] [log] [blame]
Jakub Kuderskic6c713c2023-04-21 18:36:38 -0400[diff] [blame]1################################################################################
2# Autogenerated by build_tools/bazel_to_cmake/bazel_to_cmake.py from #
3# tests/e2e/stablehlo_ops/BUILD.bazel #
4# #
5# Use iree_cmake_extra_content from iree/build_defs.oss.bzl to add arbitrary #
6# CMake-only content. #
7# #
8# To disable autogeneration for this file entirely, delete this header. #
9################################################################################
10
11iree_add_all_subdirs()
12
13iree_check_single_backend_test_suite(
14 NAME
15 check_llvm-cpu_local-task
16 SRCS
17 "abs.mlir"
18 "add.mlir"
Jakub Kuderskia10566b2023-05-08 15:40:49 -0400[diff] [blame]19 "batch_norm_inference.mlir"
Jakub Kuderskic6c713c2023-04-21 18:36:38 -0400[diff] [blame]20 "bitcast_convert.mlir"
21 "broadcast.mlir"
Jakub Kuderski4942d072023-06-01 11:14:10 -0400[diff] [blame]22 "broadcast_add.mlir"
Jakub Kuderskic6c713c2023-04-21 18:36:38 -0400[diff] [blame]23 "broadcast_in_dim.mlir"
24 "clamp.mlir"
25 "compare.mlir"
Jakub Kuderskia10566b2023-05-08 15:40:49 -0400[diff] [blame]26 "complex.mlir"
Jakub Kuderskic6c713c2023-04-21 18:36:38 -0400[diff] [blame]27 "concatenate.mlir"
28 "constant.mlir"
29 "convert.mlir"
30 "convolution.mlir"
31 "cosine.mlir"
32 "divide.mlir"
33 "dot.mlir"
34 "dot_bf16.mlir"
35 "dot_general.mlir"
36 "dynamic_slice.mlir"
37 "dynamic_update_slice.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]38 "exponential.mlir"
Jakub Kuderskif816efc2023-05-25 17:23:27 -0400[diff] [blame]39 "exponential_fp16.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]40 "exponential_minus_one.mlir"
Jakub Kuderski60197312023-05-25 16:56:59 -0400[diff] [blame]41 "fft.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]42 "finite.mlir"
43 "floor.mlir"
Jakub Kuderski5d3c2ca2023-05-25 11:36:57 -0400[diff] [blame]44 "gather.mlir"
Rob Sudermane178f452024-01-17 13:41:38 -0800[diff] [blame]45 "householder.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]46 "iota.mlir"
47 "log.mlir"
48 "log_plus_one.mlir"
49 "maximum.mlir"
50 "minimum.mlir"
51 "multiply.mlir"
52 "negate.mlir"
53 "pad.mlir"
jvstokes3fa5ad32023-06-05 15:56:00 -0700[diff] [blame]54 "philox.mlir"
Jakub Kuderskic63d2612023-05-26 14:00:14 -0400[diff] [blame]55 "pow.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]56 "reduce.mlir"
57 "reduce_window.mlir"
58 "remainder.mlir"
59 "reshape.mlir"
Jakub Kuderskie3123482023-05-01 14:01:57 -0400[diff] [blame]60 "reverse.mlir"
Jakub Kuderski109b3dd2023-05-12 15:32:06 -0400[diff] [blame]61 "rng_normal.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]62 "rng_uniform.mlir"
63 "round.mlir"
64 "rsqrt.mlir"
Jakub Kuderski472dcb92023-05-25 12:20:06 -0400[diff] [blame]65 "scatter.mlir"
Jakub Kuderskie3123482023-05-01 14:01:57 -0400[diff] [blame]66 "scatter_dynamic.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]67 "select.mlir"
68 "sine.mlir"
69 "slice.mlir"
Jakub Kuderskie3123482023-05-01 14:01:57 -0400[diff] [blame]70 "sort.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]71 "sqrt.mlir"
72 "subtract.mlir"
73 "tanh.mlir"
jvstokes3fa5ad32023-06-05 15:56:00 -0700[diff] [blame]74 "three_fry.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]75 "torch_index_select.mlir"
76 "transpose.mlir"
77 "while.mlir"
Jakub Kuderskic6c713c2023-04-21 18:36:38 -0400[diff] [blame]78 TARGET_BACKEND
79 "llvm-cpu"
80 DRIVER
81 "local-task"
Ben Vanik4f8ee512024-05-16 16:53:55 -0700[diff] [blame]82 COMPILER_FLAGS
83 "--iree-input-demote-f64-to-f32"
Stella Laurenzofd9cd2f2023-11-01 16:22:00 -0700[diff] [blame]84 INPUT_TYPE
85 "stablehlo"
Avinash Sharma30e2c202024-07-16 16:57:37 -0700[diff] [blame]86 LABELS
87 "nowasm"
Jakub Kuderskic6c713c2023-04-21 18:36:38 -0400[diff] [blame]88)
89
90iree_check_single_backend_test_suite(
91 NAME
Scott Todd78c00512024-07-10 09:37:14 -0700[diff] [blame]92 check_llvm-cpu-host_local-task
93 SRCS
94 "abs.mlir"
95 "add.mlir"
96 "batch_norm_inference.mlir"
97 "bitcast_convert.mlir"
98 "broadcast.mlir"
99 "broadcast_add.mlir"
100 "broadcast_in_dim.mlir"
101 "clamp.mlir"
102 "compare.mlir"
103 "complex.mlir"
104 "concatenate.mlir"
105 "constant.mlir"
106 "convert.mlir"
107 "convolution.mlir"
108 "cosine.mlir"
109 "divide.mlir"
110 "dot.mlir"
111 "dot_bf16.mlir"
112 "dot_general.mlir"
113 "dynamic_slice.mlir"
114 "dynamic_update_slice.mlir"
115 "exponential.mlir"
116 "exponential_fp16.mlir"
117 "exponential_minus_one.mlir"
118 "fft.mlir"
119 "finite.mlir"
120 "floor.mlir"
121 "gather.mlir"
122 "householder.mlir"
123 "iota.mlir"
124 "log.mlir"
125 "log_plus_one.mlir"
126 "maximum.mlir"
127 "minimum.mlir"
128 "multiply.mlir"
129 "negate.mlir"
130 "pad.mlir"
131 "philox.mlir"
132 "pow.mlir"
133 "reduce.mlir"
134 "reduce_window.mlir"
135 "remainder.mlir"
136 "reshape.mlir"
137 "reverse.mlir"
138 "rng_normal.mlir"
139 "rng_uniform.mlir"
140 "round.mlir"
141 "rsqrt.mlir"
142 "scatter.mlir"
143 "scatter_dynamic.mlir"
144 "select.mlir"
145 "sine.mlir"
146 "slice.mlir"
147 "sort.mlir"
148 "sqrt.mlir"
149 "subtract.mlir"
150 "tanh.mlir"
151 "three_fry.mlir"
152 "torch_index_select.mlir"
153 "transpose.mlir"
154 "while.mlir"
155 TARGET_BACKEND
156 "llvm-cpu"
157 DRIVER
158 "local-task"
159 COMPILER_FLAGS
160 "--iree-input-demote-f64-to-f32"
161 "--iree-llvmcpu-target-cpu-features=host"
162 INPUT_TYPE
163 "stablehlo"
164 LABELS
165 "hostonly"
166 "local"
167)
168
169iree_check_single_backend_test_suite(
170 NAME
Jakub Kuderskic6c713c2023-04-21 18:36:38 -0400[diff] [blame]171 check_vmvx_local-task
172 SRCS
173 "abs.mlir"
174 "add.mlir"
Jakub Kuderskia10566b2023-05-08 15:40:49 -0400[diff] [blame]175 "batch_norm_inference.mlir"
Jakub Kuderskic6c713c2023-04-21 18:36:38 -0400[diff] [blame]176 "bitcast_convert.mlir"
177 "broadcast.mlir"
Jakub Kuderski4942d072023-06-01 11:14:10 -0400[diff] [blame]178 "broadcast_add.mlir"
Jakub Kuderskic6c713c2023-04-21 18:36:38 -0400[diff] [blame]179 "broadcast_in_dim.mlir"
180 "clamp.mlir"
181 "compare.mlir"
Jakub Kuderskia10566b2023-05-08 15:40:49 -0400[diff] [blame]182 "complex.mlir"
Jakub Kuderskic6c713c2023-04-21 18:36:38 -0400[diff] [blame]183 "concatenate.mlir"
184 "constant.mlir"
185 "convert.mlir"
186 "convolution.mlir"
187 "cosine.mlir"
188 "divide.mlir"
189 "dot.mlir"
Jakub Kuderskic6c713c2023-04-21 18:36:38 -0400[diff] [blame]190 "dot_general.mlir"
191 "dynamic_slice.mlir"
192 "dynamic_update_slice.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]193 "exponential.mlir"
194 "exponential_minus_one.mlir"
Jakub Kuderski60197312023-05-25 16:56:59 -0400[diff] [blame]195 "fft.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]196 "finite.mlir"
197 "floor.mlir"
Jakub Kuderski5d3c2ca2023-05-25 11:36:57 -0400[diff] [blame]198 "gather.mlir"
Rob Sudermane178f452024-01-17 13:41:38 -0800[diff] [blame]199 "householder.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]200 "iota.mlir"
201 "log.mlir"
202 "log_plus_one.mlir"
203 "maximum.mlir"
204 "minimum.mlir"
205 "multiply.mlir"
206 "negate.mlir"
207 "pad.mlir"
jvstokes3fa5ad32023-06-05 15:56:00 -0700[diff] [blame]208 "philox.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]209 "pow.mlir"
210 "reduce.mlir"
211 "reduce_window.mlir"
212 "remainder.mlir"
213 "reshape.mlir"
Jakub Kuderskie3123482023-05-01 14:01:57 -0400[diff] [blame]214 "reverse.mlir"
Jakub Kuderski109b3dd2023-05-12 15:32:06 -0400[diff] [blame]215 "rng_normal.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]216 "rng_uniform.mlir"
217 "round.mlir"
218 "rsqrt.mlir"
Jakub Kuderski472dcb92023-05-25 12:20:06 -0400[diff] [blame]219 "scatter.mlir"
Jakub Kuderskie3123482023-05-01 14:01:57 -0400[diff] [blame]220 "scatter_dynamic.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]221 "select.mlir"
222 "sine.mlir"
223 "slice.mlir"
Jakub Kuderskie3123482023-05-01 14:01:57 -0400[diff] [blame]224 "sort.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]225 "sqrt.mlir"
226 "subtract.mlir"
227 "tanh.mlir"
jvstokes3fa5ad32023-06-05 15:56:00 -0700[diff] [blame]228 "three_fry.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]229 "torch_index_select.mlir"
230 "transpose.mlir"
231 "while.mlir"
Jakub Kuderskic6c713c2023-04-21 18:36:38 -0400[diff] [blame]232 TARGET_BACKEND
233 "vmvx"
234 DRIVER
235 "local-task"
Ben Vanik4f8ee512024-05-16 16:53:55 -0700[diff] [blame]236 COMPILER_FLAGS
237 "--iree-input-demote-f64-to-f32"
Stella Laurenzofd9cd2f2023-11-01 16:22:00 -0700[diff] [blame]238 INPUT_TYPE
239 "stablehlo"
Jakub Kuderskic6c713c2023-04-21 18:36:38 -0400[diff] [blame]240)
241
242iree_check_single_backend_test_suite(
243 NAME
244 check_vulkan-spirv_vulkan
245 SRCS
246 "abs.mlir"
247 "add.mlir"
Jakub Kuderskia10566b2023-05-08 15:40:49 -0400[diff] [blame]248 "batch_norm_inference.mlir"
Jakub Kuderskic6c713c2023-04-21 18:36:38 -0400[diff] [blame]249 "bitcast_convert.mlir"
250 "broadcast.mlir"
Jakub Kuderski4942d072023-06-01 11:14:10 -0400[diff] [blame]251 "broadcast_add.mlir"
Jakub Kuderskic6c713c2023-04-21 18:36:38 -0400[diff] [blame]252 "broadcast_in_dim.mlir"
253 "clamp.mlir"
254 "compare.mlir"
Jakub Kuderskia10566b2023-05-08 15:40:49 -0400[diff] [blame]255 "complex.mlir"
Jakub Kuderski109b3dd2023-05-12 15:32:06 -0400[diff] [blame]256 "concatenate.mlir"
Jakub Kuderskic6c713c2023-04-21 18:36:38 -0400[diff] [blame]257 "constant.mlir"
258 "convert.mlir"
259 "convolution.mlir"
260 "cosine.mlir"
261 "divide.mlir"
262 "dot.mlir"
Jakub Kuderski95c56192023-05-01 17:53:28 -0400[diff] [blame]263 "dot_bf16.mlir"
Jakub Kuderskic6c713c2023-04-21 18:36:38 -0400[diff] [blame]264 "dot_general.mlir"
265 "dynamic_slice.mlir"
266 "dynamic_update_slice.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]267 "exponential.mlir"
268 "exponential_minus_one.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]269 "finite.mlir"
270 "floor.mlir"
Jakub Kuderski5d3c2ca2023-05-25 11:36:57 -0400[diff] [blame]271 "gather.mlir"
Rob Sudermane178f452024-01-17 13:41:38 -0800[diff] [blame]272 "householder.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]273 "iota.mlir"
274 "log.mlir"
275 "log_plus_one.mlir"
276 "maximum.mlir"
277 "minimum.mlir"
278 "multiply.mlir"
279 "negate.mlir"
280 "pad.mlir"
jvstokes3fa5ad32023-06-05 15:56:00 -0700[diff] [blame]281 "philox.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]282 "pow.mlir"
283 "reduce.mlir"
284 "reduce_window.mlir"
285 "remainder.mlir"
286 "reshape.mlir"
Jakub Kuderski109b3dd2023-05-12 15:32:06 -0400[diff] [blame]287 "rng_normal.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]288 "rng_uniform.mlir"
289 "round.mlir"
290 "rsqrt.mlir"
Jakub Kuderski472dcb92023-05-25 12:20:06 -0400[diff] [blame]291 "scatter.mlir"
Jakub Kuderskie3123482023-05-01 14:01:57 -0400[diff] [blame]292 "scatter_dynamic.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]293 "select.mlir"
294 "sine.mlir"
295 "slice.mlir"
Jakub Kuderskie3123482023-05-01 14:01:57 -0400[diff] [blame]296 "sort.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]297 "sqrt.mlir"
298 "subtract.mlir"
299 "tanh.mlir"
jvstokes3fa5ad32023-06-05 15:56:00 -0700[diff] [blame]300 "three_fry.mlir"
Jakub Kuderski6852a342023-04-25 17:07:34 -0400[diff] [blame]301 "torch_index_select.mlir"
302 "transpose.mlir"
303 "while.mlir"
Jakub Kuderskic6c713c2023-04-21 18:36:38 -0400[diff] [blame]304 TARGET_BACKEND
Scott Todd78c00512024-07-10 09:37:14 -0700[diff] [blame]305 "vulkan-spirv"
Jakub Kuderskic6c713c2023-04-21 18:36:38 -0400[diff] [blame]306 DRIVER
Scott Todd78c00512024-07-10 09:37:14 -0700[diff] [blame]307 "vulkan"
Jakub Kuderskic6c713c2023-04-21 18:36:38 -0400[diff] [blame]308 COMPILER_FLAGS
Ben Vanik4f8ee512024-05-16 16:53:55 -0700[diff] [blame]309 "--iree-input-demote-f64-to-f32"
Stella Laurenzofd9cd2f2023-11-01 16:22:00 -0700[diff] [blame]310 INPUT_TYPE
311 "stablehlo"
Jakub Kuderskic6c713c2023-04-21 18:36:38 -0400[diff] [blame]312)
313
Jakub Kuderski109b3dd2023-05-12 15:32:06 -0400[diff] [blame]314iree_check_single_backend_test_suite(
315 NAME
Scott Todd6d3a7cc2024-02-02 11:16:13 -0800[diff] [blame]316 check_cuda_graph
Lei Zhang171e31c2024-01-09 19:54:53 -0500[diff] [blame]317 SRCS
318 "abs.mlir"
319 "add.mlir"
320 "batch_norm_inference.mlir"
321 "bitcast_convert.mlir"
322 "broadcast.mlir"
323 "broadcast_add.mlir"
324 "broadcast_in_dim.mlir"
325 "clamp.mlir"
326 "compare.mlir"
327 "complex.mlir"
328 "concatenate.mlir"
329 "constant.mlir"
330 "convert.mlir"
331 "convolution.mlir"
332 "cosine.mlir"
333 "divide.mlir"
334 "dot.mlir"
335 "dot_bf16.mlir"
336 "dot_general.mlir"
337 "dynamic_slice.mlir"
338 "dynamic_update_slice.mlir"
339 "exponential.mlir"
340 "exponential_fp16.mlir"
341 "exponential_minus_one.mlir"
342 "fft.mlir"
343 "finite.mlir"
344 "floor.mlir"
345 "gather.mlir"
Rob Sudermane178f452024-01-17 13:41:38 -0800[diff] [blame]346 "householder.mlir"
Lei Zhang171e31c2024-01-09 19:54:53 -0500[diff] [blame]347 "iota.mlir"
348 "log.mlir"
349 "log_plus_one.mlir"
350 "maximum.mlir"
351 "minimum.mlir"
352 "multiply.mlir"
353 "negate.mlir"
354 "pad.mlir"
355 "philox.mlir"
356 "pow.mlir"
357 "reduce.mlir"
358 "reduce_window.mlir"
359 "remainder.mlir"
360 "reshape.mlir"
361 "reverse.mlir"
362 "rng_normal.mlir"
363 "rng_uniform.mlir"
364 "round.mlir"
365 "rsqrt.mlir"
366 "scatter.mlir"
367 "scatter_dynamic.mlir"
368 "select.mlir"
369 "sine.mlir"
370 "slice.mlir"
371 "sort.mlir"
372 "sqrt.mlir"
373 "subtract.mlir"
374 "tanh.mlir"
375 "three_fry.mlir"
376 "torch_index_select.mlir"
377 "transpose.mlir"
378 "while.mlir"
379 TARGET_BACKEND
380 "cuda"
381 DRIVER
Lei Zhang3b3cef92024-01-23 14:11:36 -0800[diff] [blame]382 "cuda"
Lei Zhang171e31c2024-01-09 19:54:53 -0500[diff] [blame]383 COMPILER_FLAGS
Ben Vanik4f8ee512024-05-16 16:53:55 -0700[diff] [blame]384 "--iree-input-demote-f64-to-f32"
Lei Zhang171e31c2024-01-09 19:54:53 -0500[diff] [blame]385 "--iree-stream-emulate-memset"
Lei Zhang171e31c2024-01-09 19:54:53 -0500[diff] [blame]386 INPUT_TYPE
387 "stablehlo"
388 RUNNER_ARGS
Scott Todd6d3a7cc2024-02-02 11:16:13 -0800[diff] [blame]389 "--cuda_use_streams=false"
Lei Zhang171e31c2024-01-09 19:54:53 -0500[diff] [blame]390 LABELS
391 "noasan"
392 "nomsan"
393 "notsan"
394 "noubsan"
395 "requires-gpu-nvidia"
396)
397
398iree_check_single_backend_test_suite(
399 NAME
Scott Todd6d3a7cc2024-02-02 11:16:13 -0800[diff] [blame]400 check_cuda_stream
Lei Zhang171e31c2024-01-09 19:54:53 -0500[diff] [blame]401 SRCS
402 "abs.mlir"
403 "add.mlir"
404 "batch_norm_inference.mlir"
405 "bitcast_convert.mlir"
406 "broadcast.mlir"
407 "broadcast_add.mlir"
408 "broadcast_in_dim.mlir"
409 "clamp.mlir"
410 "compare.mlir"
411 "complex.mlir"
412 "concatenate.mlir"
413 "constant.mlir"
414 "convert.mlir"
415 "convolution.mlir"
416 "cosine.mlir"
417 "divide.mlir"
418 "dot.mlir"
419 "dot_bf16.mlir"
420 "dot_general.mlir"
421 "dynamic_slice.mlir"
422 "dynamic_update_slice.mlir"
423 "exponential.mlir"
424 "exponential_fp16.mlir"
425 "exponential_minus_one.mlir"
426 "fft.mlir"
427 "finite.mlir"
428 "floor.mlir"
429 "gather.mlir"
Rob Sudermane178f452024-01-17 13:41:38 -0800[diff] [blame]430 "householder.mlir"
Lei Zhang171e31c2024-01-09 19:54:53 -0500[diff] [blame]431 "iota.mlir"
432 "log.mlir"
433 "log_plus_one.mlir"
434 "maximum.mlir"
435 "minimum.mlir"
436 "multiply.mlir"
437 "negate.mlir"
438 "pad.mlir"
439 "philox.mlir"
440 "pow.mlir"
441 "reduce.mlir"
442 "reduce_window.mlir"
443 "remainder.mlir"
444 "reshape.mlir"
445 "reverse.mlir"
446 "rng_normal.mlir"
447 "rng_uniform.mlir"
448 "round.mlir"
449 "rsqrt.mlir"
450 "scatter.mlir"
451 "scatter_dynamic.mlir"
452 "select.mlir"
453 "sine.mlir"
454 "slice.mlir"
455 "sort.mlir"
456 "sqrt.mlir"
457 "subtract.mlir"
458 "tanh.mlir"
459 "three_fry.mlir"
460 "torch_index_select.mlir"
461 "transpose.mlir"
462 "while.mlir"
463 TARGET_BACKEND
464 "cuda"
465 DRIVER
Lei Zhang3b3cef92024-01-23 14:11:36 -0800[diff] [blame]466 "cuda"
Ben Vanik4f8ee512024-05-16 16:53:55 -0700[diff] [blame]467 COMPILER_FLAGS
468 "--iree-input-demote-f64-to-f32"
Lei Zhang171e31c2024-01-09 19:54:53 -0500[diff] [blame]469 INPUT_TYPE
470 "stablehlo"
471 RUNNER_ARGS
Scott Todd6d3a7cc2024-02-02 11:16:13 -0800[diff] [blame]472 "--cuda_use_streams=true"
Lei Zhang171e31c2024-01-09 19:54:53 -0500[diff] [blame]473 LABELS
474 "noasan"
475 "nomsan"
476 "notsan"
477 "noubsan"
478 "requires-gpu-nvidia"
479)
480
Scott Todd78c00512024-07-10 09:37:14 -0700[diff] [blame]481iree_check_single_backend_test_suite(
482 NAME
483 check_rocm_hip_stream
484 SRCS
485 "abs.mlir"
486 "add.mlir"
487 "batch_norm_inference.mlir"
488 "bitcast_convert.mlir"
489 "broadcast.mlir"
490 "broadcast_add.mlir"
491 "broadcast_in_dim.mlir"
492 "clamp.mlir"
493 "compare.mlir"
494 "complex.mlir"
495 "concatenate.mlir"
496 "constant.mlir"
497 "convert.mlir"
498 "convolution.mlir"
499 "cosine.mlir"
500 "divide.mlir"
501 "dot.mlir"
502 "dot_bf16.mlir"
503 "dot_general.mlir"
504 "dynamic_slice.mlir"
505 "dynamic_update_slice.mlir"
506 "exponential.mlir"
507 "exponential_fp16.mlir"
508 "exponential_minus_one.mlir"
509 "fft.mlir"
510 "finite.mlir"
511 "floor.mlir"
512 "gather.mlir"
513 "householder.mlir"
514 "iota.mlir"
515 "log.mlir"
516 "log_plus_one.mlir"
517 "maximum.mlir"
518 "minimum.mlir"
519 "multiply.mlir"
520 "negate.mlir"
521 "pad.mlir"
522 "philox.mlir"
523 "pow.mlir"
524 "reduce.mlir"
525 "reduce_window.mlir"
526 "remainder.mlir"
527 "reshape.mlir"
528 "reverse.mlir"
529 "rng_normal.mlir"
530 "rng_uniform.mlir"
531 "round.mlir"
532 "rsqrt.mlir"
533 "scatter.mlir"
534 "scatter_dynamic.mlir"
535 "select.mlir"
536 "sine.mlir"
537 "slice.mlir"
538 "sort.mlir"
539 "sqrt.mlir"
540 "subtract.mlir"
541 "tanh.mlir"
542 "three_fry.mlir"
543 "torch_index_select.mlir"
544 "transpose.mlir"
545 "while.mlir"
546 TARGET_BACKEND
547 "rocm"
548 DRIVER
549 "hip"
550 INPUT_TYPE
551 "stablehlo"
552 RUNNER_ARGS
553 "--hip_use_streams=true"
554)
555
556iree_check_single_backend_test_suite(
557 NAME
Scott Todd6145b652024-07-30 13:16:50 -0700[diff] [blame^]558 check_metal-spirv_metal
Scott Todd78c00512024-07-10 09:37:14 -0700[diff] [blame]559 SRCS
560 "abs.mlir"
561 "add.mlir"
562 "batch_norm_inference.mlir"
563 "bitcast_convert.mlir"
564 "broadcast.mlir"
565 "broadcast_add.mlir"
566 "broadcast_in_dim.mlir"
567 "clamp.mlir"
568 "compare.mlir"
569 "complex.mlir"
570 "concatenate.mlir"
571 "constant.mlir"
572 "convert.mlir"
573 "convolution.mlir"
574 "cosine.mlir"
575 "divide.mlir"
576 "dot.mlir"
577 "dot_bf16.mlir"
578 "dot_general.mlir"
579 "dynamic_slice.mlir"
580 "dynamic_update_slice.mlir"
581 "exponential.mlir"
582 "exponential_fp16.mlir"
583 "exponential_minus_one.mlir"
584 "fft.mlir"
585 "finite.mlir"
586 "floor.mlir"
587 "gather.mlir"
588 "householder.mlir"
589 "iota.mlir"
590 "log.mlir"
591 "log_plus_one.mlir"
592 "maximum.mlir"
593 "minimum.mlir"
594 "multiply.mlir"
595 "negate.mlir"
596 "pad.mlir"
597 "philox.mlir"
598 "pow.mlir"
599 "reduce.mlir"
600 "remainder.mlir"
601 "reshape.mlir"
602 "reverse.mlir"
603 "rng_normal.mlir"
604 "rng_uniform.mlir"
605 "round.mlir"
606 "rsqrt.mlir"
607 "scatter.mlir"
608 "scatter_dynamic.mlir"
609 "select.mlir"
610 "sine.mlir"
611 "slice.mlir"
612 "sort.mlir"
613 "sqrt.mlir"
614 "subtract.mlir"
615 "tanh.mlir"
616 "three_fry.mlir"
617 "torch_index_select.mlir"
618 "transpose.mlir"
619 "while.mlir"
620 TARGET_BACKEND
621 "metal-spirv"
622 DRIVER
623 "metal"
624 INPUT_TYPE
625 "stablehlo"
626)
627
Jakub Kuderskic6c713c2023-04-21 18:36:38 -0400[diff] [blame]628### BAZEL_TO_CMAKE_PRESERVES_ALL_CONTENT_BELOW_THIS_LINE ###
Jakub Kuderski4942d072023-06-01 11:14:10 -0400[diff] [blame]629
630iree_check_single_backend_test_suite(
631 NAME
632 check_webgpu
633 SRCS
634 "abs.mlir"
635 "add.mlir"
636 "batch_norm_inference.mlir"
637 "bitcast_convert.mlir"
638 "broadcast.mlir"
639 "broadcast_add.mlir"
640 "broadcast_in_dim.mlir"
641 # "clamp.mlir" # TODO(#10906): fix (i8/i16?)
642 # "compare.mlir" # TODO(#10906): fix (i8/i16?)
643 # "complex.mlir" # TODO(#11054)
644 "concatenate.mlir"
645 "constant.mlir"
646 # "convert.mlir" # TODO(#10906): fix (i8/i16?)
647 "convolution.mlir"
648 "cosine.mlir"
649 "divide.mlir"
650 "dot.mlir"
651 "dot_general.mlir"
652 "dynamic_slice.mlir"
653 "dynamic_update_slice.mlir"
654 "exponential.mlir"
655 "exponential_fp16.mlir"
656 "exponential_minus_one.mlir"
657 # "fft.mlir" # TODO(#9583): fix (fft codegen via spirv)
658 # "finite.mlir" # TODO(#11321): error: value cannot be represented as 'f32': inf
659 "floor.mlir"
660 "gather.mlir"
661 "iota.mlir"
662 "log.mlir"
663 "log_plus_one.mlir"
664 # "maximum.mlir" # TODO(#10906): fix (i8/i16?)
665 # "minimum.mlir" # TODO(#10906): fix (i8/i16?)
666 "multiply.mlir"
667 "negate.mlir"
668 "pad.mlir"
jvstokes3fa5ad32023-06-05 15:56:00 -0700[diff] [blame]669 # "philox.mlir" # TODO(#12509): WebGPU SPIR-V broken
Jakub Kuderski4942d072023-06-01 11:14:10 -0400[diff] [blame]670 "pow.mlir"
671 "reduce.mlir"
672 "reduce_window.mlir"
673 "remainder.mlir"
674 "reshape.mlir"
675 "reverse.mlir"
676 "rng_normal.mlir"
677 "rng_uniform.mlir"
678 "round.mlir"
679 "rsqrt.mlir"
680 "scatter.mlir"
681 "scatter_dynamic.mlir"
682 "select.mlir"
683 "sine.mlir"
684 "slice.mlir"
685 "sort.mlir"
686 "sqrt.mlir"
687 "subtract.mlir"
688 "tanh.mlir"
jvstokes3fa5ad32023-06-05 15:56:00 -0700[diff] [blame]689 # "three_fry.mlir" # TODO(#12509): WebGPU SPIR-V broken
Jakub Kuderski4942d072023-06-01 11:14:10 -0400[diff] [blame]690 "torch_index_select.mlir"
691 "transpose.mlir"
692 # "while.mlir" # TODO(#12509): WebGPU SPIR-V broken
693 TARGET_BACKEND
Ben Vanikeeda5ca2024-02-27 14:18:32 -0800[diff] [blame]694 "webgpu-spirv"
Jakub Kuderski4942d072023-06-01 11:14:10 -0400[diff] [blame]695 # Only test compilation for now, the WebGPU driver is not stable/tested yet.
696 # DRIVER
697 # "webgpu"
698 COMPILER_FLAGS
699 "--iree-input-type=stablehlo"
700 "--iree-codegen-gpu-native-math-precision=true" # TODO(#11321): Infer/flip default
701)
702
Lei Zhang446f96e2023-10-24 19:41:43 -0700[diff] [blame]703iree_check_single_backend_test_suite(
704 NAME
Scott Todd5ccf2dc2023-11-28 13:24:07 -0800[diff] [blame]705 check_rocm-rocm
706 SRCS
707 "abs.mlir"
708 "add.mlir"
709 "batch_norm_inference.mlir"
710 "bitcast_convert.mlir"
711 "broadcast.mlir"
712 "broadcast_add.mlir"
713 "broadcast_in_dim.mlir"
714 "clamp.mlir"
715 "compare.mlir"
716 "complex.mlir"
717 "concatenate.mlir"
718 "constant.mlir"
719 "convert.mlir"
720 "convolution.mlir"
721 "cosine.mlir"
722 "divide.mlir"
723 "dot.mlir"
724 "dot_bf16.mlir"
725 "dot_general.mlir"
726 "dynamic_slice.mlir"
727 "dynamic_update_slice.mlir"
728 "exponential.mlir"
729 "exponential_fp16.mlir"
730 "exponential_minus_one.mlir"
731 "fft.mlir"
732 "finite.mlir"
733 "floor.mlir"
734 "gather.mlir"
735 "iota.mlir"
736 "log.mlir"
737 "log_plus_one.mlir"
738 "maximum.mlir"
739 "minimum.mlir"
740 "multiply.mlir"
741 "negate.mlir"
742 "pad.mlir"
743 "philox.mlir"
744 "pow.mlir"
745 "reduce.mlir"
746 "reduce_window.mlir"
747 "remainder.mlir"
748 "reshape.mlir"
749 "reverse.mlir"
750 "rng_normal.mlir"
751 "rng_uniform.mlir"
752 "round.mlir"
753 "rsqrt.mlir"
754 "scatter.mlir"
755 "scatter_dynamic.mlir"
756 "select.mlir"
757 "sine.mlir"
758 "slice.mlir"
759 "sort.mlir"
760 "sqrt.mlir"
761 "subtract.mlir"
762 "tanh.mlir"
763 "three_fry.mlir"
764 "torch_index_select.mlir"
765 "transpose.mlir"
766 "while.mlir"
767 TARGET_BACKEND
768 "rocm"
769 # Only test compilation for now, the ROCm driver is experimental.
770 # DRIVER
771 # "rocm"
772 COMPILER_FLAGS
773 "--iree-input-type=stablehlo"
774)