class_docstr: Can be used in place of real tensors when calling:
      kernel.warmup(MockTensor(torch.float32), ...)
class_docstr_tok: ["Can","be","used","in","place","of","real","tensors","when","calling",":","kernel.warmup","(","MockTensor","(","torch.float32",")",",","...",")"]
code: def wrap_dtype(arg):
      if (
          arg.__class__.__name__ == "dtype"
          and arg.__module__ == "torch"
      ):
          return MockTensor(arg)
      return arg
  
imports: from __future__ import annotations, division
  import ast
  import hashlib
  import inspect
  import os
  import textwrap
  from collections import defaultdict, namedtuple
  from functools import cached_property
  from typing import Callable, Generic, Iterable, List, Optional, TypeVar, Union, cast, overload
  from ..runtime.driver import driver
label_desc: Use image node_id 1 for calling the MockTensor obj's underlying member method code with example usage: obj.wrap_dtype(arg) and returns: arg, MockTensor