gpu.launch_func: add accessors to grid configuration operands
authorAlex Zinenko <zinenko@google.com>
Tue, 25 Jun 2019 11:21:10 +0000 (04:21 -0700)
committerA. Unique TensorFlower <gardener@tensorflow.org>
Tue, 25 Jun 2019 16:18:47 +0000 (09:18 -0700)
commit43c0cf109ffcec12bb59d1cb5452c66ab074983a
tree992d36c5a5bda0f295d6b10745ce0ca9bf6cea6f
parent14a10a818b2d069afe209a8f247aed8e4dbca014
gpu.launch_func: add accessors to grid configuration operands

Add accessor functions that return `gpu::KernelDim3` containing the respective
operands for grid and block size accepted by `gpu.launch_func`.  Use the same
signature as for `gpu.launch`.

PiperOrigin-RevId: 254942674
mlir/include/mlir/GPU/GPUDialect.h
mlir/lib/GPU/IR/GPUDialect.cpp