code: def compile_fn(attrs, capability):
      @triton.jit
      def kernel_sub(a, b, o, N: tl.constexpr):
          idx = tl.arange(0, N)
          tl.store(o + idx, tl.load(a + idx) - tl.load(b + idx) * 777)
  
      src = ASTSource(
          fn=kernel_sub,
          constants={3: 32},
          signature={0: "*fp32", 1: "*fp32", 2: "*fp32"},
          attrs=attrs,
      )
      triton.compile(src=src, target=("cuda", capability))
  
file_path: triton/python/test/unit/runtime/test_subproc.py
imports: import multiprocessing
  import os
  import shutil
  import torch
  import triton
  import triton.language
  from triton.compiler import ASTSource
label_desc: Use image node_id 2 for calling a global function with example usage: compile_fn(attrs, capability) without return types