code: def test_fp8_unary_op(np_dtype, dtype_str):
      func = fp8_unary(dtype_str)
      if not tvm.testing.device_enabled("llvm"):
          return
      if np_dtype is None:
          """Skip test if ml_dtypes is not installed"""
          return
  
      f = tvm.build(func, target="llvm")
      a = np.random.randn(128).astype(np_dtype)
      b = np.random.randn(128).astype(np_dtype)
      a_add_b = np.zeros(128).astype(np_dtype)
      a_sub_b = np.zeros(128).astype(np_dtype)
      a_mul_b = np.zeros(128).astype(np_dtype)
      a_fp32 = np.zeros(128).astype(np.float32)
      a_roundtrip = np.zeros(128).astype(np_dtype)
      args = list(
          map(
              lambda _: tvm.nd.array(_),
              [a, b, a_add_b, a_sub_b, a_mul_b, a_fp32, a_roundtrip],
          )
      )
      f(*args)
  
file_path: tvm/tests/python/ir/test_datatype_nv_fp8.py
imports: import numpy
  import tvm
  import tvm.testing
  import tvm.tir
  from tvm import te
  from tvm.script import tir
label_desc: Use image node_id 3 for calling a global function with example usage: test_fp8_unary_op(np_dtype, dtype_str) without return types