I have a kernel where in one particular line I define a tensor of -inf. This is all in the body of a function decorated with triton.jit, so it's not clear to me why this triton.jit issue is raised.
m_i = tl.zeros([BLOCK_M], dtype=tl.float32) - float("inf")
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/conda/envs/triton/lib/python3.11/site-packages/triton/runtime/jit.py", line 801, in __call__
raise RuntimeError("Cannot call @triton.jit'd outside of the scope of a kernel")
RuntimeError: Cannot call @triton.jit'd outside of the scope of a kernel
Versions:
I have a kernel where in one particular line I define a tensor of -inf. This is all in the body of a function decorated with triton.jit, so it's not clear to me why this triton.jit issue is raised.
Error trace