[mlir][gpu] Add subgroup Id/Size/Num to GPU dialect
authorThomas Raoux <thomasraoux@google.com>
Thu, 4 Jun 2020 17:45:53 +0000 (10:45 -0700)
committerThomas Raoux <thomasraoux@google.com>
Thu, 4 Jun 2020 17:52:40 +0000 (10:52 -0700)
commit661235e1267c389a7d56023da7fcb036468ecfda
tree05d57291b28c4ffd20f8a2f5084354d04f1ac6e2
parente53f558057846760c77f5f2b9967104516a207d7
[mlir][gpu] Add subgroup Id/Size/Num to GPU dialect

Add SubgroupId, SubgroupSize and NumSubgroups to GPU dialect ops and add the
lowering of those ops to SPIRV.

Differential Revision: https://reviews.llvm.org/D81042
mlir/include/mlir/Dialect/GPU/GPUOps.td
mlir/lib/Conversion/GPUToSPIRV/ConvertGPUToSPIRV.cpp
mlir/lib/Dialect/SPIRV/SPIRVLowering.cpp
mlir/test/Conversion/GPUToSPIRV/builtins.mlir
mlir/test/Dialect/GPU/ops.mlir