code: def test_load_reduce(BLOCK_M, BLOCK_N, dtype_str):
      dtype = dtype_mapping[dtype_str]
      x = torch.randn((BLOCK_M, BLOCK_N), device="cuda", dtype=dtype)
      y = torch.empty((BLOCK_M,), device="cuda", dtype=dtype)
  
      load_reduce_kernel[(1,)](
          x, y, x.stride(0), x.stride(1), y.stride(0), BLOCK_M, BLOCK_N
      )
  
      golden = x.max(dim=1)[0]
      torch.set_printoptions(profile="full")
      assert_close(y, golden, 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 4 for calling a global function with example usage: test_load_reduce(BLOCK_M, BLOCK_N, dtype_str) without return types