[Mlir-commits] [mlir] [NVGPU] Fix nvdsl examples (PR #156830)

[LLVM Continuous Integration]

llvm-ci wrote:

LLVM Buildbot has detected a new failure on builder `mlir-nvidia` running on `mlir-nvidia` while building `mlir` at step 7 "test-build-check-mlir-build-only-check-mlir".

Full details are available at: https://lab.llvm.org/buildbot/#/builders/138/builds/21474

<details>
<summary>Here is the relevant piece of the build log for the reference</summary>

```
Step 7 (test-build-check-mlir-build-only-check-mlir) failure: test (failure)
******************** TEST 'MLIR :: Examples/NVGPU/Ch2.py' FAILED ********************
Exit Code: 2

Command Output (stdout):
--
# RUN: at line 1
env SUPPORT_LIB=/vol/worker/mlir-nvidia/mlir-nvidia/llvm.obj/lib/libmlir_cuda_runtime.so  sh -c 'if [[ "%mlir_run_cuda_sm90_tests" == "1" ]];  then /usr/bin/python3.10 /vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/Ch2.py | /vol/worker/mlir-nvidia/mlir-nvidia/llvm.obj/bin/FileCheck /vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/Ch2.py;  else export MLIR_NVDSL_PRINT_IR=1;  /usr/bin/python3.10 /vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/Ch2.py | /vol/worker/mlir-nvidia/mlir-nvidia/llvm.obj/bin/FileCheck /vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/Ch2.py --check-prefix=DUMPIR; fi'
# executed command: env SUPPORT_LIB=/vol/worker/mlir-nvidia/mlir-nvidia/llvm.obj/lib/libmlir_cuda_runtime.so sh -c 'if [[ "%mlir_run_cuda_sm90_tests" == "1" ]];  then /usr/bin/python3.10 /vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/Ch2.py | /vol/worker/mlir-nvidia/mlir-nvidia/llvm.obj/bin/FileCheck /vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/Ch2.py;  else export MLIR_NVDSL_PRINT_IR=1;  /usr/bin/python3.10 /vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/Ch2.py | /vol/worker/mlir-nvidia/mlir-nvidia/llvm.obj/bin/FileCheck /vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/Ch2.py --check-prefix=DUMPIR; fi'
# .---command stderr------------
# | sh: 1: [[: not found
# | Traceback (most recent call last):
# |   File "/vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/Ch2.py", line 90, in <module>
# |     saxpy(x, y, alpha)
# |   File "/vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/tools/nvdsl.py", line 426, in wrapper
# |     result = funcBody(*fargs, **kwargs)
# |   File "/vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/Ch2.py", line 35, in saxpy
# |     t1 = gpu.wait(token_ty, [])
# | TypeError: wait() takes from 0 to 1 positional arguments but 2 were given
# | FileCheck error: '<stdin>' is empty.
# | FileCheck command line:  /vol/worker/mlir-nvidia/mlir-nvidia/llvm.obj/bin/FileCheck /vol/worker/mlir-nvidia/mlir-nvidia/llvm.src/mlir/test/Examples/NVGPU/Ch2.py --check-prefix=DUMPIR
# `-----------------------------
# error: command failed with exit status: 2

--

********************


```

</details>

https://github.com/llvm/llvm-project/pull/156830