code: def matmul(
      lhs, rhs, transa=False, transb=False, n=0, m=0, dtype=None
  ):
      """Create an extern op that compute matrix mult of A and rhs with cuBLAS
  
      Parameters
      ----------
      lhs : Tensor
          The left matrix operand
      rhs : Tensor
          The right matrix operand
      transa : bool
          Whether transpose lhs
      transb : bool
          Whether transpose rhs
  
      Returns
      -------
      C : Tensor
          The result tensor.
      """
      if n == 0:
          n = lhs.shape[1] if transa else lhs.shape[0]
      if m == 0:
          m = rhs.shape[0] if transb else rhs.shape[1]
      dtype = dtype if dtype is not None else lhs.dtype
      return te.extern(
          (n, m),
          [lhs, rhs],
          lambda ins, outs: tvm.tir.call_packed(
              "tvm.contrib.cublaslt.matmul",
              ins[0],
              ins[1],
              outs[0],
              transa,
              transb,
          ),
          dtype=dtype,
          name="C",
      )
  
label_desc: Use image node_id 1 for calling a global function with example usage: matmul(lhs, rhs, transa, transb, n, m, dtype) and returns: te