How to execute string code with @triton.jit in python

[starstream]

Describe the bug

source_code = """
@triton.jit
def add_kernel():
   ...
"""
exec(source_code)

size = 1024
x = torch.rand(size, device="cuda")
y = torch.rand(size, device="cuda")
add_kernel[grid](x, y, output, size, BLOCK_SIZE=1024)

Traceback (most recent call last):
File "", line 3, in
File "/opt/conda/lib/python3.10/site-packages/triton/runtime/jit.py", line 546, in jit
return decorator(fn)
File "/opt/conda/lib/python3.10/site-packages/triton/runtime/jit.py", line 537, in decorator
return JITFunction(
File "/opt/conda/lib/python3.10/site-packages/triton/runtime/jit.py", line 419, in init
self.starting_line_number = inspect.getsourcelines(fn)[1]
File "/opt/conda/lib/python3.10/inspect.py", line 1121, in getsourcelines
lines, lnum = findsource(object)
File "/opt/conda/lib/python3.10/inspect.py", line 958, in findsource
raise OSError('could not get source code')
OSError: could not get source code

Environment details

Triton 3.1.0
GPU NvidiaA100

[knwng]

Maybe you can take a look at patch_kernel.