[mlir][spirv] Query target environment for mapping memory space
authorStanley Winata <stanley@nod-labs.com>
Tue, 20 Sep 2022 22:28:40 +0000 (15:28 -0700)
committerStanley Winata <stanley@nod-labs.com>
Tue, 20 Sep 2022 22:28:58 +0000 (15:28 -0700)
commit1dc48a916a1bbf99799dbeefef26d9078e159e93
tree28c745d35996e7962cf1e6e3a047bd3af3814aa7
parent079a5ffb0a3f862bee3076f785d5c71c03e14f00
[mlir][spirv] Query target environment for mapping memory space

Checks spirv::TargetEnv from op to see if it contains either Kernel or Shader capabilities.
If it does, then it will set the memory space mapping accordingly.

Reviewed By: antiagainst

Differential Revision: https://reviews.llvm.org/D134317
mlir/lib/Conversion/MemRefToSPIRV/MapMemRefStorageClassPass.cpp
mlir/test/Conversion/MemRefToSPIRV/map-storage-class.mlir