[mlir] accept values with result numbers in gpu.launch_func
authorAlex Zinenko <zinenko@google.com>
Mon, 16 Jan 2023 15:01:02 +0000 (15:01 +0000)
committerAlex Zinenko <zinenko@google.com>
Mon, 16 Jan 2023 19:26:42 +0000 (19:26 +0000)
commit3f2f83ef41419507bcdaf751b86713ef193b7de0
treeaa519a8ccad266a28a60c8e7b5d250ac327aed72
parent9954516ffb10a39ae0c8517f4865eec24e07d56e
[mlir] accept values with result numbers in gpu.launch_func

The parser of gpu.launch_func was incorrectly rejecting SSA values with
result numbers (`%0#0`) in the list of function arguments by using the
`parseArgument` function intended for region argument declarations, not
operands. Fix this by directly parsing comma-separated operands and
types.

Reviewed By: nicolasvasilache

Differential Revision: https://reviews.llvm.org/D141851
mlir/lib/Dialect/GPU/IR/GPUDialect.cpp
mlir/test/Dialect/GPU/ops.mlir