code: def test_add(SIZE, BLOCK_SIZE, dtype_str):
      dtype = dtype_mapping[dtype_str]
      output = torch.empty(SIZE, device="cuda", dtype=dtype)
      x = torch.randn(SIZE, device="cuda", dtype=dtype)
      y = torch.randn(SIZE, device="cuda", dtype=dtype)
  
      def grid(meta):
          return (triton.cdiv(SIZE, meta["BLOCK_SIZE"]),)
  
      add_kernel[grid](x, y, output, SIZE, BLOCK_SIZE=BLOCK_SIZE)
  
      output_torch = x + y
      torch.set_printoptions(profile="full")
      assert_close(
          output, output_torch, rtol=1e-2, atol=1e-3, check_dtype=False
      )
  
file_path: triton/python/test/unit/hopper/test_mixed_io.py
imports: import pytest
  import torch
  from torch.testing import assert_close
  import triton
  import triton.language
label_desc: Use image node_id 2 for calling a global function with example usage: test_add(SIZE, BLOCK_SIZE, dtype_str) and returns: triton