code: def test_annotation():
      @triton.jit
      def kernel(X, i: tl.int32):
          tl.store(X, i)
  
      x = torch.empty(1, dtype=torch.int32, device="cuda")
  
      device = torch.cuda.current_device()
      kernel[(1,)](x, 1)
      kernel[(1,)](x, 8)
      kernel[(1,)](x, 16)
      kernel[(1,)](x, 17)
      assert len(kernel.cache[device]) == 4
  
imports: import importlib.util
  import os
  import shutil
  import tempfile
  import pytest
  import torch
  import triton
  import triton.language
  from triton.runtime.jit import JITFunction
label_desc: Use image node_id 14 for calling a global function with example usage: test_annotation() without return types