Autotune encountered an error when enabling the debug function

[wangshankun]

Tutorials example matmul：

TRITON_INTERPRET=1 python 03-matrix-multiplication.py

Encountered an error：

matmul_kernel() got an unexpected keyword argument 'warmup'

Error log：

Traceback (most recent call last): File "/root/triton/python/tutorials/03-matrix-multiplication.py", line 309, in triton_output = matmul(a, b) File "/root/triton/python/tutorials/03-matrix-multiplication.py", line 288, in matmul matmul_kernel[grid]( File "/usr/local/lib/python3.10/dist-packages/triton-2.1.0-py3.10-linux-x86_64.egg/triton/runtime/jit.py", line 166, in return lambda *args, kwargs: self.run(grid=grid, warmup=False, *args, *kwargs) File "/usr/local/lib/python3.10/dist-packages/triton-2.1.0-py3.10-linux-x86_64.egg/triton/runtime/autotuner.py", line 144, in run timings = {config: self._bench(args, config=config, kwargs) for config in pruned_configs} File "/usr/local/lib/python3.10/dist-packages/triton-2.1.0-py3.10-linux-x86_64.egg/triton/runtime/autotuner.py", line 144, in timings = {config: self._bench(*args, config=config, kwargs) for config in pruned_configs} File "/usr/local/lib/python3.10/dist-packages/triton-2.1.0-py3.10-linux-x86_64.egg/triton/runtime/autotuner.py", line 123, in _bench return do_bench(kernel_call, warmup=self.num_warmups, rep=self.num_reps, quantiles=(0.5, 0.2, 0.8)) File "/usr/local/lib/python3.10/dist-packages/triton-2.1.0-py3.10-linux-x86_64.egg/triton/testing.py", line 100, in do_bench fn() File "/usr/local/lib/python3.10/dist-packages/triton-2.1.0-py3.10-linux-x86_64.egg/triton/runtime/autotuner.py", line 110, in kernel_call self.fn.run( File "/usr/local/lib/python3.10/dist-packages/triton-2.1.0-py3.10-linux-x86_64.egg/triton/runtime/interpreter.py", line 533, in run return GridExecutor(self.fn, self.arg_names, grid)(*args, *kwargs) File "/usr/local/lib/python3.10/dist-packages/triton-2.1.0-py3.10-linux-x86_64.egg/triton/runtime/interpreter.py", line 500, in call args = inspect.getcallargs(self.fn, args_hst, kwargs) File "/usr/lib/python3.10/inspect.py", line 1517, in getcallargs raise TypeError("%s() got an unexpected keyword argument %r" % TypeError: matmul_kernel() got an unexpected keyword argument 'warmup'

[chen-yingfa]

Hello, I am faced with the same error, have you found any solutions?

[Jokeren]

Please install triton nightly