[mlir][spirv] Respect client API requirements for 64-bit index
authorLei Zhang <antiagainst@google.com>
Mon, 27 Feb 2023 06:15:18 +0000 (06:15 +0000)
committerLei Zhang <antiagainst@google.com>
Mon, 27 Feb 2023 06:16:50 +0000 (06:16 +0000)
commit9a4c768a7d83158d225a2c9baa82ee5d827fb726
treee6c30f076f6010d4dfda061cafedc5887dd411e1
parent2aa12583e6ace921a1e58f39d54a35296273758e
[mlir][spirv] Respect client API requirements for 64-bit index

Vulkan requires GPU processor ID/count builtin variables to be
32-bit scalar or vector for all the cases. Similarly there
are special requirements for OpenCL. We need to make sure those
rules are respected when converting using 64bit for index.

Reviewed By: kuhar

Differential Revision: https://reviews.llvm.org/D144819
mlir/lib/Conversion/GPUToSPIRV/GPUToSPIRV.cpp
mlir/test/Conversion/GPUToSPIRV/builtins.mlir