diff options
| author | Matthias Springer <springerm@google.com> | 2022-10-27 12:24:02 +0200 |
|---|---|---|
| committer | Matthias Springer <springerm@google.com> | 2022-10-27 12:29:57 +0200 |
| commit | 09dfb44193973db76f33e6518b83e53a762cfb34 (patch) | |
| tree | febe309fed6bc38e721b4c23f4a86ba76f8a4213 /llvm/lib/IR/BasicBlock.cpp | |
| parent | e26f28711c1f81b7f74ce749269e7539492d3934 (diff) | |
| download | llvm-09dfb44193973db76f33e6518b83e53a762cfb34.zip llvm-09dfb44193973db76f33e6518b83e53a762cfb34.tar.gz llvm-09dfb44193973db76f33e6518b83e53a762cfb34.tar.bz2 | |
[mlir][tensor][bufferize] Support memory_space for tensor.pad
This change adds memory space support to tensor.pad. (tensor.generate and tensor.from_elements do not support memory spaces yet.)
The memory space is inferred from the buffer of the source tensor.
Instead of lowering tensor.pad to tensor.generate + tensor.insert_slice, it is now lowered to bufferization.alloc_tensor (with the correct memory space) + linalg.map + tensor.insert_slice.
Memory space support for the remaining two tensor ops is left for a later point, as this requires some more design discussions.
Differential Revision: https://reviews.llvm.org/D136265
Diffstat (limited to 'llvm/lib/IR/BasicBlock.cpp')
0 files changed, 0 insertions, 0 deletions