|
1 | | -# linalg dialect to gpu dialect lowering pipeline |
2 | | -# Ready for vulkan runner or narrow scope l0/sycl runner starting from GPU dialect. |
3 | | -builtin.module(convert-tensor-to-linalg |
| 1 | +// linalg dialect to gpu dialect lowering pipeline |
| 2 | +// Ready for vulkan runner or narrow scope l0/sycl runner starting from GPU dialect. |
| 3 | +builtin.module( |
| 4 | + convert-tensor-to-linalg |
4 | 5 | arith-bufferize |
5 | | - func.func(empty-tensor-to-alloc-tensor |
6 | | - eliminate-empty-tensors |
7 | | - scf-bufferize |
8 | | - shape-bufferize |
9 | | - linalg-bufferize |
10 | | - bufferization-bufferize |
11 | | - tensor-bufferize) |
| 6 | + func.func( |
| 7 | + empty-tensor-to-alloc-tensor |
| 8 | + eliminate-empty-tensors |
| 9 | + scf-bufferize |
| 10 | + shape-bufferize |
| 11 | + linalg-bufferize |
| 12 | + bufferization-bufferize |
| 13 | + tensor-bufferize) |
12 | 14 | func-bufferize |
13 | | - func.func(finalizing-bufferize |
14 | | - convert-linalg-to-parallel-loops |
15 | | - gpu-map-parallel-loops |
16 | | - convert-parallel-loops-to-gpu) |
17 | | -# insert-gpu-allocs pass can have client-api = opencl or vulkan args |
| 15 | + func.func( |
| 16 | + finalizing-bufferize |
| 17 | + convert-linalg-to-parallel-loops |
| 18 | + gpu-map-parallel-loops |
| 19 | + convert-parallel-loops-to-gpu |
| 20 | + ) |
| 21 | +// insert-gpu-allocs pass can have client-api = opencl or vulkan args |
18 | 22 | func.func(insert-gpu-allocs) |
19 | 23 | canonicalize |
20 | 24 | normalize-memrefs |
21 | | -# Unstride memrefs does not seem to be needed. |
22 | | -# func.func(unstride-memrefs) |
| 25 | +// Unstride memrefs does not seem to be needed. |
| 26 | +// func.func(unstride-memrefs) |
23 | 27 | func.func(lower-affine) |
24 | 28 | gpu-kernel-outlining |
25 | 29 | canonicalize |
26 | 30 | cse |
27 | | -# The following set-spirv-* passes can have client-api = opencl or vulkan args |
| 31 | + |
| 32 | +// The following set-spirv-* passes can have client-api = opencl or vulkan args |
28 | 33 | set-spirv-capabilities |
29 | 34 | gpu.module(set-spirv-abi-attrs) |
30 | | - canonicalize) |
31 | | -# End |
| 35 | + canonicalize |
| 36 | + ) |
| 37 | +// End |
0 commit comments